Folder structure of ./data.
./data
├── CharadesSTA
│ ├── VGG
│ │ └── vgg_rgb_features.hdf5
│ ├── C3D
│ │ └── Charades_C3D.hdf5
│ ├── I3D
│ │ └── features
│ │ └── video_id.npy
│ ├── train.json
│ ├── test.json
│ └── multi_test.json
│
├── ActivityNet
│ ├── C3D
│ │ └── activitynet_v1-3_c3d.hdf5
│ ├── I3D
│ │ └── video_id.npy
│ ├── train.json
│ ├── val.json
│ ├── test.json
│ └── multi_test.json
│
└── QVHighlights
├── features
│ ├── clip_features
│ ├── clip_text_features
│ └── slowfast_features
├── train.json
├── val.json
└── test.json
- Install python packages.
pip install -r requirements.txt
- (Optional) Boost mAP calculation.
python setup.py install
-
Single GPU training.
python main.py --config path/to/config.json --logdir path/to/log/dir -
Multi-GPU training.
CUDA_VISIBLE_DEVICES=0,1,2,3 python main.py --config path/to/config.json --logdir path/to/log/dir
- Testing best.pth in the logdir.
python main.py --test_only --config path/to/config.json --logdir path/to/log/dir
tensorboard --logdir path/to/log