C3D进行视频分类：使用自己数据集

一. 代码框架

1. 整体代码的框架

2、dataloader的代码框架

2.1 def init()的框架

 

2.2 def getitem()的框架

 

二、修改数据集

1.源代码使用的数据集是UCF101，目录结构为ucf101-results/train/类别/类别_g0*_c0*/图片，仿照该目录结构修改自己的数据

2. 注意：每一帧图片的大小进行修改，这一部分的代码在make_train_test.py中，图片为112*112。因为在生成224*224的图片后，cuda内存不足，训练器无法运算

三、训练器中需要修改的地方

 1. dataset的名字以及num_classes

2. save_dir_root以及save_dir

3. 在加载数据集部分，train_loader  train_size部分

4. 写入tensorboard部分，writer.add_scalar部分的地址

 

最后，有报错不要急，慢慢修改嘛，修改也是学习的过程呀