百度论文复现3D-ResNets,原项目地址:https://github.com/kenshohara/3D-resnets-pytorch
AI Studio 地址:https://aistudio.baidu.com/aistudio/projectdetail/735326 数据集地址:https://aistudio.baidu.com/aistudio/datasetdetail/48916
- Python 3+
- paddlepaddle 1.8.0 +
pip install -r requirement.txt
1.视频转图片
python generate_video_jpgs.py <video_dir> <pic_dir> ucf101
2.训练
python train.py
可开启MIX-UP数据增强,效果未验证。
python train.py --mixup
3.测试结果
python test.py
python eval_accuracy.py
top-1准确率为93.55% 略高于论文中的92.9%。