Prerequisites

• Python 3.6
• Pytorch 0.4
• CUDA 9.0

Installation the environment

Please first refer to MattNet to prepare related data.

Generate the Triads

python ./tools/prepro_rel.py --dataset refcoco --splitBy unc

Training

First stage

CUDA_VISIBLE_DEVICES=1 python ./tools/twophrase_train.py --dataset refcoco --splitBy unc --exp_id 1

second stage

CUDA_VISIBLE_DEVICES=1 python ./tools/twophrase_train_strong.py --dataset refcoco --splitBy unc --exp_id 1

Evaluation

First stage

CUDA_VISIBLE_DEVICES=1 python ./tools/twophrase_eval.py --dataset refcoco --splitBy unc --split val --id 1

seconda stage

CUDA_VISIBLE_DEVICES=1 python ./tools/twophrase_eval_strong.py --dataset refcoco --splitBy unc --split val --id 1