方法出自论文 ECCV2018 " Efficient Convolutional Network for Online Video Understanding"
接口出自https://github.com/mzolfaghari/ECO-pytorch

环境要求

作者:
Python 3.5.2
PyTorch 0.4.1
TorchVison: 0.2.1

我的:
centos系统
Python 3.6.7 Anaconda
PyTorch 1.0.0
TorchVison: 0.2.1
通过命令可以直接安装PyTorch TorchVison,这个需要看不同人的系统和环境配置,可以在pytorch官网找到相应的命令

conda install pytorch torchvision cuda100 -c pytorch

下载源码

git clone https://github.com/mzolfaghari/ECO-pytorch

下载ucf数据集

地址:http://crcv.ucf.edu/data/UCF101/UCF101.rar
将数据集解压后放在ECO-pytorch/dataset/目录下

制作数据集

因为下载下来的数据集是视频avi格式,因此我们需要将每个视频转成图像帧集。作者并没有提供对应的代码,可以自己写一个,或者找到TSN项目中tools目录下的build_of.py,并在此基础上删除生成光流的部分。
利用build_of.py得到新的图像帧的数据集,具体目录如下:
PyTorch implementation for "ECO,finetune on ucf101
每一个文件夹对应一个视频,文件夹里存放视频的图像帧

然后利用命令生成训练的训练文件

python gen_dataset_lists.py ucf101 ./dataset/ucf101/

在data目录下得到文件
PyTorch implementation for "ECO,finetune on ucf101

下载预训练模型

作者提供了预训练模型,可以在百度云上下载,下载后无需解压,放在models目录下

训练

通过脚本文件进行训练,需要注意的是,训练前修改相应的文件名和路径

    ./scripts/run_ECOLite_finetune_UCF101.sh local

相关文章:

  • 2021-07-16
  • 2022-12-23
  • 2022-12-23
  • 2022-01-15
  • 2022-12-23
  • 2022-01-27
  • 2021-04-19
  • 2021-08-06
猜你喜欢
  • 2022-12-23
  • 2021-12-14
  • 2021-08-27
  • 2022-12-23
  • 2021-04-23
  • 2021-11-10
  • 2021-08-20
相关资源
相似解决方案