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.