/SlowFastNetworks

PyTorch implementation of "SlowFast Networks for Video Recognition".

Primary LanguagePython

SlowFastNetworks

PyTorch implementation of "SlowFast Networks for Video Recognition".

Train

  1. Dataset should be orgnized as:
dataset(e.g. UCF-101)  
│    │ train/training  
│    │    │ ApplyEyeMakeup  
│    │    │ ApplyLipstick  
│    │    │ ...  
│    │ validation  
     │    │ ApplyEyeMakeup  
     │    │ ApplyLipstick  
     │    │ ...   
  1. Modify the params in config.py and mode of train_dataloader or val_dataloader in train.py.

Requirements

python 3
PyTorch >= 0.4.1
tensorboardX
OpenCV

Code Reference:

[1] https://github.com/Guocode/SlowFast-Networks/
[2] https://github.com/jfzhang95/pytorch-video-recognition
[3] https://github.com/irhumshafkat/R2Plus1D-PyTorch