欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-03-05 09:05:29
...

opts.py

  1. 添加一个必填参数 root_path ,在后面的 dataset.py 文件中定义 TSNDataSet 类 文件中需要用到。
    行为识别之 TSN 部分代码修改下图是 dataset.py文件中TSNDataSet类使用到的地方(源码无需修改)
    行为识别之 TSN 部分代码修改行为识别之 TSN 部分代码修改

main.py

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

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

运行代码如下(根据自己的配置修改 –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