Official implementation for Hierarchical Deep Residual Reasoning for Temporal Moment Localization
We use the PyTorch
framework.
- Python version: 3.6
- PyTorch version: 1.6
Clone the repository:
git clone https://github.com/ddlBoJack/HDRR.git
cd HDRR
We preprocess the data in the same way as MIGCN do.
Once you get the data above, you need to do semantic role labeling using allennlp
:
cd code
python semantic_role_labeling.py
Train HDRR on Charades-STA/ActivityNet with various features:
python main.py
Test HDRR on Charades-STA/ActivityNet with various features:
python main.py --test --model_load_path
List other hyper-parameters by:
python main.py -h
Please cite the following paper if HDRR is helpful for your research.
@article{ma2021hierarchical,
title={Hierarchical Deep Residual Reasoning for Temporal Moment Localization},
author={Ma, Ziyang and Han, Xianjing and Song, Xuemeng and Cui, Yiran and Nie, Liqiang},
journal={arXiv preprint arXiv:2111.00417},
year={2021}
}