DVD-GAN
This repo is an implementation of Efficient Video Generation on Complex Datasets
Prerequisite
Package | version |
---|---|
python | >=3.5 |
pytorch | 1.12 |
numpy | 1.17.2 |
pandas | 0.25.1 |
tensorboardX | 1.8 |
ffmpeg | 3.4.2 |
Note: For more detail, please look up requirements.txt
Prepare datasets
sudo apt install ffmpeg # important package
chmod u+x scripts/data_prepare.sh
scripts/data_prepare.sh <dataset_path>
Train the model
scripts/train_model.sh <runing_mode> <dataset_path>
Dataset
Process UCF-101
- Step 1: Download dataset
- Step 2: Convert from avi to jpg files using: python utils/video_jpg_ucf101_hmdb51.py avi_video_directory jpg_video_directory
- Step 3: Generate n_frames files using: python utils/n_frames_ucf101_hmdb51.py jpg_video_directory
- Step 4: Generate annotation file in json format similar to ActivityNet using: python utils/ucf101_json.py annotation_dir_path
Note: To change the number of class:
- Modify classInd.txt to contain the expected class(es). For example: 1 ApplyEyeMakeup 2 ApplyLipstick 3 Archery
- Run step 4 only
- The code in dataloader automatically skips the unsed videos.