TDN复现(UCF101--只用了十个类别)

1.参照c3d配置mmaction2环境,基本参照这一篇文章完成复现

https://blog.csdn.net/KalutSirocco/article/details/123758916?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522170001744016800182789186%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=170001744016800182789186&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduend~default-1-123758916-null-null.142^v96^pc_search_result_base6&utm_term=TDN%E9%83%A8%E7%BD%B2&spm=1018.2226.3001.4187

 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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值