source github: https://github.com/jwyang/faster-rcnn.pytorch/tree/pytorch-1.0
cd FASTER_RCNN
pip install -r requirements.txt
rm lib/build/*
cd lib & python setup.py build develop
python demo.py \
--dataset [DATASET] \
--net res101 \
--load_dir results/run \
--pretrained_model_name faster_rcnn.pth \
--cuda \
--vis \
--image_dir [FOLDER-TO-TEST] \
Dataset options: ricoCustomized, rico2k, rico10k, ricoDefault, ricoText
Put the pretrained model in the folder FASTER_RCNN/results/run/res101/[DATASET]
For more information, see https://github.com/jwyang/faster-rcnn.pytorch/tree/pytorch-1.0