Skip to content

MachineLP/PyTorch_video_classifier

Repository files navigation

ref: https://github.com/jfzhang95/pytorch-video-recognition

ref:https://github.com/dmlc/gluon-cv

Installation

The code was tested with Anaconda and Python 3.5. After installing the Anaconda environment:

  1. Clone the repo:

    git clone https://github.com/MachineLP/Pytorch_video_classifier.git
    cd Pytorch_video_classifier
  2. Install dependencies:

    For PyTorch dependency, see pytorch.org for more details.

    For custom dependencies:

    conda install opencv
    pip install tqdm scikit-learn tensorboardX
  3. Download pretrained model from BaiduYun or GoogleDrive. Currently only support pretrained model for C3D.

  4. Configure your dataset and pretrained model path in mypath.py.

  5. You can choose different models and datasets in train.py.

    To train the model, please do:

    python train.py
  6. inference To infer the model, please do:

    python inference.py

About

video recognition

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages