This repo is the implementation of paper Weakly-Supervised Action Localization via Embedding-Modeling Iterative Optimization.
- Ubuntu 16.04.6
- Python 3.6
- Cuda 9.0
- PyTorch 1.1.0
- Install dependencies:
pip install -r requirements.txt. - Prepare THUMOS14 and ActivityNet datasets.
We employ I3D features in the paper.
We recommend to extract the features using the followingf repo:
- For training the model:
python train.py
- For testing the model:
python test.py
The final results are saved in .npz format.
If you have any questions, please contact me (shihaichao@iie.ac.cn).
MIT