KaihuaTang / Kinetics-Dataloader

A pytorch kinetics-400/kinetics-600 dataloader

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

A self-contained pytorch Kinetics Dataloader

This project provides a compact self-contained kinetics dataloader based on Pytorch. The codes are modified from SlowFast.

Environments

Run the following commands to install the exvironments

conda create -n YOUR_ENV_NAME pip python=3.8
source activate YOUR_ENV_NAME
conda install pytorch==1.6.0 torchvision==0.7.0 -c pytorch
conda install pillow
conda install matplotlib
pip install pyyaml
pip install av

Prepare Datasets

Follow the Dataset Preprocessing to prepare the Kinetics-400/Kinetics-600 datasets.

Demo

Run the Demo to visualize the sampled video frames. This project keep the most compact version of dataloader by removing all the frame-level data augmentation.

About

A pytorch kinetics-400/kinetics-600 dataloader

License:GNU General Public License v3.0


Languages

Language:Jupyter Notebook 91.2%Language:Python 8.8%