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