Temporal Segment Network (TSN) 是视频分类领域经典的基于2D-CNN的解决方案。该方法主要解决视频的长时间行为判断问题，通过稀疏采样视频帧的方式代替稠密采样，既能捕获视频全局信息，也能去除冗余，降低计算量。最终将每帧特征平均融合后得到视频的整体特征，并用于分类。本代码实现的模型为基于单路RGB图像的TSN网络结构，Backbone采用ResNet-50结构。
详细内容请参考ECCV 2016年论文Temporal Segment Networks: Towards Good Practices for Deep Action Recognition
TSN的训练数据采用由DeepMind公布的Kinetics-400动作识别数据集。数据下载及准备请参考数据说明
数据准备完毕后，可以通过如下两种方式启动训练：
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
export FLAGS_fast_eager_deletion_mode=1
export FLAGS_eager_delete_tensor_gb=0.0
export FLAGS_fraction_of_gpu_memory_to_use=0.98
python train.py --model_name=TSN \
--config=./configs/tsn.yaml \
--log_interval=10 \
--valid_interval=1 \
--use_gpu=True \
--save_dir=./data/checkpoints \
--fix_random_seed=False \
--pretrain=$PATH_TO_PRETRAIN_MODEL
bash run.sh train TSN ./configs/tsn.yaml
-
从头开始训练，需要加载在ImageNet上训练的ResNet50权重作为初始化参数，请下载此模型参数并解压，将上面启动命令行或者run.sh脚本中的
pretrain参数设置为解压之后的模型参数 存放路径。如果没有手动下载并设置pretrain参数，则程序会自动下载并将参数保存在~/.paddle/weights/ResNet50_pretrained目录下面 -
可下载已发布模型model通过
--resume指定权重存 放路径进行finetune等开发
数据读取器说明： 模型读取Kinetics-400数据集中的mp4数据，每条数据抽取seg_num段，每段抽取1帧图像，对每帧图像做随机增强后，缩放至target_size。
训练策略：
- 采用Momentum优化算法训练，momentum=0.9
- 权重衰减系数为1e-4
- 学习率在训练的总epoch数的1/3和2/3时分别做0.1的衰减
可通过如下两种方式进行模型评估:
python eval.py --model_name=TSN \
--config=./configs/tsn.yaml \
--log_interval=1 \
--weights=$PATH_TO_WEIGHTS \
--use_gpu=True
bash run.sh eval TSN ./configs/tsn.yaml
-
使用
run.sh进行评估时，需要修改脚本中的weights参数指定需要评估的权重 -
若未指定
--weights参数，脚本会下载已发布模型model进行评估 -
评估结果以log的形式直接打印输出TOP1_ACC、TOP5_ACC等精度指标
-
使用CPU进行评估时，请将上面的命令行或者run.sh脚本中的
use_gpu设置为False
当取如下参数时，在Kinetics400的validation数据集下评估精度如下:
| seg_num | target_size | Top-1 |
|---|---|---|
| 3 | 224 | 0.66 |
| 7 | 224 | 0.67 |
可通过如下两种方式启动模型推断：
python predict.py --model_name=TSN \
--config=./configs/tsn.yaml \
--log_interval=1 \
--weights=$PATH_TO_WEIGHTS \
--filelist=$FILELIST \
--use_gpu=True \
--video_path=$VIDEO_PATH
bash run.sh predict TSN ./configs/tsn.yaml
-
使用
run.sh进行评估时，需要修改脚本中的weights参数指定需要用到的权重。 -
如果video_path为'', 则忽略掉此参数。如果video_path != ''，则程序会对video_path指定的视频文件进行预测，而忽略掉filelist的值，预测结果为此视频的分类概率。
-
若未指定
--weights参数，脚本会下载已发布模型model进行推断 -
模型推断结果以log的形式直接打印输出，可以看到测试样本的分类预测概率。
-
使用CPU进行推断时，请将命令行或者run.sh脚本中的
use_gpu设置为False
- Temporal Segment Networks: Towards Good Practices for Deep Action Recognition, Limin Wang, Yuanjun Xiong, Zhe Wang, Yu Qiao, Dahua Lin, Xiaoou Tang, Luc Van Gool