YOWO框架支持几种主流的数据集包括UCF101-24,AVA等,在之前的文章中已经介绍过UCF101-24复现过程及效果,这篇文章介绍个人复现AVA数据集的效果。
一、数据集准备
AVA数据集下载过程非常简单,但是准备过程十分繁琐,读者可以通过以下链接下载AVA数据集并着手开始准备。
https://github.com/yjh0410/AVA_Datasethttps://github.com/yjh0410/AVA_Dataset
下载完成之后,依次运行step1,step2,step3,step4的sh文件,Linux系统用户可以直接运行,Windows用户运行sh文件方法请见另一篇博客。
二、代码修改
如同另一篇博客(UCF101-24)修改代码的方法,我们需要在dataset/ava.py以及train.py文件内修改。


修改完成后,可以直接运行train.py或者通过终端命令行运行,终端命令如下:
python train.py --cuda

本文介绍了如何使用YOWO框架处理AV阿数据集,包括数据集的准备步骤,代码的相应修改,以及作者自行训练得到的效果,与官方结果相近但有微小提升。

2640

被折叠的 条评论
为什么被折叠?



