asrafulashiq / wsad

Weakly Supervised Temporal Action Localization Using Deep Metric Learning

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

Weakly Supervised Temporal Action Localization Using Deep Metric Learning


Languages

Language:Python 100.0%