Skip to content

Latest commit

 

History

History
32 lines (21 loc) · 893 Bytes

README.md

File metadata and controls

32 lines (21 loc) · 893 Bytes
Raw

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