行为识别之 TSN 部分代码修改

opts.py

  1. 添加一个必填参数 root_path ,在后面的 dataset.py 文件中定义 TSNDataSet 类 文件中需要用到。
    opts.py 文件修改部分(增加新选项)下图是 dataset.py文件中TSNDataSet类使用到的地方(源码无需修改)
    传入参数 root_path 使用 传入参数 root_path

main.py

  1. train()与validate()函数中重复代码 ,修改如图
    修改1
    参数对应于 opts.py 文件中定义好的
    1.ucf101 对应 dataset
    2.RGB 对应 modality
    3./datasets/ucf101/rawframes/ 对应于自己定义的root_path
    4.其他参数依次类推

    参考图

运行代码如下(根据自己的配置修改 –num_segments 之前的三个与目录相关的地址)

python main.py ucf101 RGB /datasets/ucf101/rawframes/ /datasets/ucf101/ucf101_train_split_1_rawframes.txt /datasets/ucf101/ucf101_val_split_1_rawframes.txt --num_segments 3 --gd 20 --lr 0.001 --lr_steps 30 60 --epoch 5 -b 16 -j 8 --dropout 0.8 --snapshot_pref ucf101_bninception_

代码解读 及其他修改 可以参考博客

训练及测试代码解读 参考 [ 1 ] [ 2 ] 两篇博客
环境搭建参考 github 源码 或 博客 [ 3 ]
[1]: https://blog.csdn.net/u014380165/article/details/79058147
[2]: https://blog.csdn.net/u014380165/article/details/79058381
[3]: https://blog.csdn.net/qq_39862223/article/details/108461526

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值