【资源】视频研究常用方法、数据集和任务汇总

作者建立了一个用于视频研究领域常用库,包括常用方法、数据集和任务等,非常全面,分享给大家~
作者:gsig
GitHub地址:https://github.com/gsig/PyVideoResearch

PyVideoResearch

  • A repositsory of common methods, datasets, and tasks for video research
  • Please note that this repository is in the process of being released to the public. Please bear with us as we standardize the API and streamline the code.
  • Some of the baselines were run with an older version of the codebase (but the git commit hash is available for each experiment) and might need to be updated.
  • We encourage you to submit a Pull Request to help us document and incorporate as many baselines and datasets as possible to this codebase
  • We hope this project will be of value to the community and everyone will consider adding their methods to this codebase

List of implemented methods:

  • I3D
  • 3D ResNet
  • Asynchronous Temporal Fields
  • Actor Observer Network
  • Temporal Segment Networks
  • Temporal Relational Networks
  • Non-local neural networks
  • Two-Stream Networks
  • I3D Mask-RCNN
  • 3D ResNet Video Autoencoder



List of supported datasets:

  • Charades
  • CharadesEgo
  • Kinetics
  • AVA
  • ActivityNet
  • Something Something
  • Jester



List of supported tasks:

  • Action classification
  • Action localization
  • Spatial Action localization
  • Inpainting
  • Video Alignment
  • Triplet Classification



Contributor: Gunnar Atli Sigurdsson

  • If this code helps your research, please consider citing:
@inproceedings{sigurdsson2018pyvideoresearch,
author = {Gunnar A. Sigurdsson and Abhinav Gupta},
title = {PyVideoResearch},
year={2018},
code = {https://github.com/gsig/PyVideoResearch},
}


Installation Instructions

Requirements:

  • Python 2.7 or Python 3.6
  • PyTorch 0.4 or PyTorch 1.0

Python packages:

  • numpy
  • ffmpeg-python
  • PIL
  • cv2
  • torchvision



See external libraries under external/ for requirements if using their corresponding baselines.

Run the following to get both this repository and the remote repositories under external/

git clone git@github.com:gsig/PyVideoResearch.git
git submodule update --init --recursive


Steps to train your own network:

  • Download the corresponding dataset
  • Duplicate and edit one of the experiment files under exp/ with appropriate parameters. For additional parameters, see opts.py
  • Run an experiment by calling python exp/rgbnet.py where rgbnet.py is your experiment file. See baseline_exp/ for a variety of baselines.
  • The checkpoints/logfiles/outputs are stored in your specified cache directory.
  • Build of the code, cite our papers, and say hi to us at CVPR.

Good luck!

Pretrained networks:

We are in the process of preparing and releasing the pre-trained models. If anything is missing, please let us know. The names correspond to experiments under "baseline_exp". While we standardize the names, please be aware that some of the model may have names listed after "original name" in the experiment file. We also provide the generated log.txt file for each experiment as name.txt

The models are stored here: https://www.dropbox.com/sh/duodxydolzz5qfl/AAC0i70lv8ssVRWg4ux5Vv9pa?dl=0


ResNet50 pre-trained on Charades

  • resnet50_rgb.pth.tar
  • resnet50_rgb_python3.pth.tar

ResNet1010 pre-trained on Charades

  • resnet101_rgb.pth.tar
  • resnet101_rgb_python3.pth.tar

I3D pre-trained on ImageNet (courtesy of https://github.com/piergiaj)

  • aj_rgb_imagenet.pth

I3D pre-trained on ImageNet+Kinetics (courtesy of https://github.com/piergiaj)

  • aj_rgb_kinetics.pth

actor_observer_3d_charades_ego.py

actor_observer_charades_ego.py

actor_observer_classification_charades_ego.py

async_tf_i3d_charades.py

  • async__par1.pth.tar
  • async__par1.txt

i3d_ava.py

i3d_mask_rcnn_ava.py

i3d_something_something.py

inpainting.py

nonlocal_resnet50_3d_charades.py

nonlocal_resnet50_3d_kinetics.py

  • i3d12b2.pth.tar
  • i3d12b2.txt

resnet50_3d_charades.py

  • i3d12b2.pth.tar
  • i3d12b2.txt

resnet50_3d_kinetics.py

  • i3d8k.pth.tar
  • i3d8k.txt

temporal_relational_networks_charades.py

temporal_relational_networks_something_something.py

  • trn4b.pth.tar
  • trn4b.txt

temporal_segment_networks_activity_net.py

temporal_segment_networks_charades.py

  • trn2f3b.pth.tar
  • trn2f3b.txt

two_stream_kinetics.py

two_stream_networks_activity_net.py

  • anet2.pth.tar
  • anet2.txt





推荐阅读
更侧重工业的应用:PRCV2018 美图短视频实时分类挑战赛冠军技术方案
谷歌提出新方法:基于单目视频的无监督深度学习结构化
干货 | OpenCV 中 KLT 光流跟踪原理详解与代码演示