Weakly Supervised Temporal Action Localization Using Deep Metric Learning

This repository contains Pytorch code for our paper Weakly Supervised Temporal Action Localization Using Deep Metric Learning which has been accepted to WACV 2020.

Installation

This package requires Pytorch 1.4. Other Pytorch versions should also work. You can create environment using conda:

conda env create -n wsad -f environment.yml
conda activate wsad

Download Dataset

Download Thumos14-reduced dataset from here.

Training

python main.py

Testing

python main.py --test --ckpt [pretrained weight]

Citation

If you find this work useful, please cite our work:

@misc{islam2020weakly,
    title={Weakly Supervised Temporal Action Localization Using Deep Metric Learning},
    author={Ashraful Islam and Richard J. Radke},
    year={2020},
    eprint={2001.07793},
    archivePrefix={arXiv},
    primaryClass={cs.CV}
}

Acknowledgments

sujoyp/wtalc-pytorch