GAN_Mask-RCNN

Please refer to the follwing links for installation intructions and usage:

• maskrcnn-benchmark: facebookresearch/maskrcnn-benchmark
• PreciseRoiPooling: vacancy/PreciseRoIPooling

Multi-GPU training

We follow the facebook-benchmark command for multi-gpu training. For more information on how to change the config file for different number of gpus, please check the maskrcnn-benchmark link above. (Note that our config is designed to work on 2 GPUs)

export NGPUS=2
python -m torch.distributed.launch --nproc_per_node=$NGPUS train.py --config-file "configs/e2e_mask_rcnn_R_101_FPN_1x_phone.yaml"