st-resnet
This repository contains the code for our NIPS'16 and CVPR'17 papers:
Christoph Feichtenhofer, Axel Pinz, Richard P. Wildes "Spatiotemporal Residual Networks for Video Action Recognition" in Proc. NIPS 2016 Christoph Feichtenhofer, Axel Pinz, Richard P. Wildes "Spatiotemporal Multiplier Networks for Video Action Recognition" in Proc. CVPR 2017
If you use our code/models/data for your research, please cite our papers:
@inproceedings{feichtenhofer2016spatiotemporal, title={Spatiotemporal residual networks for video action recognition}, author={Feichtenhofer, Christoph and Pinz, Axel and Wildes, Richard}, booktitle={Advances in Neural Information Processing Systems (NIPS)}, pages={3468--3476}, year={2016} } @inproceedings{feichtenhofer2017multiplier, title={Spatiotemporal multiplier networks for video action recognition}, author={Feichtenhofer, Christoph and Pinz, Axel and Wildes, Richard P} booktitle={Conference on Computer Vision and Pattern Recognition (CVPR)}, year={2017} }
The code was tested on Ubuntu 14.04, 16.04 and Windows 10 using MATLAB R2016b and NVIDIA Titan X GPUs.
If you have questions regarding the implementation please contact:
Christoph Feichtenhofer <feichtenhofer AT tugraz.at>
================================================================================
Download the code git clone --recursive https://github.com/feichtenhofer/st-resnet
Compile the code by running compile.m
.
This will also compile our own branch of the MatConvNet toolbox. In case of any issues, please follow the installation instructions on the MatConvNet homepage.
Edit the file cnn_setup_environment.m to adjust the models and data paths.
(Optional) Download pretrained model files and the datasets, linked below and unpack them into your models/data directory. Otherwise the scripts will attempt to download the models at runtime.
Optionally you can also pretrain your own twostream base models by running
base_streams/cnn_ucf101_spatial();
to train the appearance network streams.
base_streams/cnn_ucf101_temporal();
to train the optical flow network streams.
Training STResNet_stage1();
, STResNet_stage2();
to train the architecture in our NIPS 2016 paper. STMulNet();
to train the architecture in our CVPR 2017 paper.
In case you did not download or trained the base models, the script will attempt to download these accordingly.
In case you would like to train on the CPU, clear the variable opts.train.gpus
In case you encounter memory issues on your GPU, consider decreasing the cudnnWorkspaceLimit
(512MB is default)
Download final models here: (in case you do not, STResNet_stage1();
, STResNet_stage2();
, and STResNet_test();
will attempt to download the respective models at runtime.)
Download final models here: (in case you do not, STMulNet();
and STMulNet_test();
will attempt to download the models at runtime.)
Pre-computed optical flow images and resized rgb frames for the UCF101 and HMDB51 datasets
Our Optical flow extraction tool provides OpenCV wrappers for optical flow extraction on a GPU.
还没有评论,说两句吧!
热门资源
Keras-ResNeXt
Keras ResNeXt Implementation of ResNeXt models...
seetafaceJNI
项目介绍 基于中科院seetaface2进行封装的JAVA...
spark-corenlp
This package wraps Stanford CoreNLP annotators ...
capsnet-with-caps...
CapsNet with capsule-wise convolution Project ...
inferno-boilerplate
This is a very basic boilerplate example for pe...
智能在线
400-630-6780
聆听.建议反馈
E-mail: support@tusaishared.com