1.参照c3d配置mmaction2环境,基本参照这一篇文章完成复现
2.问题
(1)配置环境
pytorch1.7.0有bug,别用,2.0版本太高,还有模型加载出错问题,选择了A4000未出错,且加载很快,其他卡加载有问题,且没找到解决方法
(2)记录文件顺序以及修改点(路径均为绝对路径)
(3)校验SSL链接报错
在miniconda3/lib/python3.8/urlib/request.py里面添加这两句,就不会报错
import ssl
ssl._create_default_https_context = ssl._create_unverified_context
(4)训练模型:轮数使用命令来指定,此条指定了150epoch
python -m torch.distributed.launch --master_port 12347 --nproc_per_node=1 main.py ucf101 RGB --arch resnet50 --num_segments 1 --gd 20 --lr 0.01 --lr_scheduler step --lr_steps 30 45 55 --epochs 150 --batch-size 8 --wd 5e-4 --dropout 0.5 --consensus_type=avg --eval-freq=1 -j 4 --npb
(5)测试模型
创建pkl文件夹
python test_models_center_crop.py ucf101 --archs='resnet50' --weights /root/autodl-tmp/mmaction2/TDN/checkpoint/TDN__ucf101_RGB_resnet50_avg_segment1_e150/best.pth.tar --test_segments=1 --test_crops=1 --batch_size=16 --gpus 0 --output_dir /root/autodl-tmp/mmaction2/TDN/pkl -j 0 --clip_index=0
(6)查看总体结果
python pkl_to_results.py --num_clips 1 --test_crops 1 --output_dir /root/autodl-tmp/mmaction2/TDN/pkl