当前位置:网站首页>百度飞将BMN时序动作定位框架 | 数据准备与训练指南 (下)

百度飞将BMN时序动作定位框架 | 数据准备与训练指南 (下)

2022-07-06 18:06:00 訢詡

三、BMN

上篇说到处理一下BMN需要的训练数据,在PaddleVideo-develop/applications/FootballAction/datasets/EuroCup2016/features这个文件下面

(1)训练数据处理

打开PaddleVideo-develop/applications/FootballAction/datasets/script这个目录,在这个路径下面执行:

python get_instance_for_bmn.py

(2)修改配置文件

打开PaddleVideo-develop/applications/FootballAction/train_proposal/configs/bmn_football_v2.0.yaml

第22、26、31行改为刚才生成的:PaddleVideo-develop/applications/FootballAction/datasets/EuroCup2016/input_for_bmn/label.json的绝对路径

第38、49、60行改为datasets绝对路径:PaddleVideo-develop/applications/FootballAction/datasets/EuroCup2016/input_for_bmn/feature

第84行PaddleVideo-develop/applications/FootballAction/BMN_INFERENCE_results的绝对路径然后在这个位置新建一个叫做BMN_INFERENCE_results的文件夹

第17行batchsize改为1

(3)训练BMN

在PaddleVideo项目根目录执行:

python -B -m paddle.distributed.launch --gpus="0" --log_dir=./football/logs_bmn main.py  --validate -c applications/FootballAction/train_proposal/configs/bmn_football_v2.0.yaml -o output_dir=./football/bmn

开始训练

 

原网站

版权声明
本文为[訢詡]所创,转载请带上原文链接,感谢
https://blog.csdn.net/Andrwin/article/details/125632317