Awesome
Prerequisites
- Python 2.7
- Pytorch 0.4
- CUDA 9.0
Installation the environment
Please first refer to MattNet to prepare related data.
Pre-trained models
All pre-trained models and related data can be downloaded here.
Apart from the triads generated by the parsing mechasnism, we also provide the human-annotated triads for all queries for future works, whihc can be downloaded here.
Generate the Triads
python ./tools/prepro_rel.py --dataset refcoco --splitBy unc
Training
python ./tools/train.py --dataset refcoco --splitBy unc --exp_id 1
Evaluation
python ./eval.py --dataset refcoco --splitBy unc --split val --id 1