行为识别之 TSN 部分代码修改
程序员文章站
2022-03-05 09:05:29
...
opts.py
-
添加一个必填参数 root_path ,在后面的 dataset.py 文件中定义 TSNDataSet 类 文件中需要用到。
下图是 dataset.py文件中TSNDataSet类使用到的地方(源码无需修改)
main.py
-
train()与validate()函数中重复代码 ,修改如图
参数对应于 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
上一篇: linux安装php运行环境