fatalfeel/faster_rcnn_pytorch

Faster-Rcnn in Pytorch

• An implementation of Faster R-CNN in PyTorch.
• Only python source. These is no need to compile nms and roialign cpp
• Comment in many functions
• You can debug any line in cpu mode

Prepare install

• cd faster_rcnn_pytorch
• ./install_data.sh

Run

• python3 ./train.py --cuda True
• python3 ./train.py --cuda True --resume True
• python3 ./infer.py --cuda True

Performance

• GeForce GTX 1650 4GB
• CUDA version 10.2
• Resnet-101
• Train 2.3 frames per second
• Infer 5 frames per second

Pretrained download

• https://drive.google.com/file/d/1dPoxbqVq3RswQx7T4nn19_qEHnMK_0Ep
• Copy to ./checkpoint

Compare and learn

You can compare with simple-faster-rcnn that can debug any line in cpu mode

• https://www.mediafire.com/file/m5wx06gmtqvu1km/simple-faster-rcnn.tar.gz

Web site

• http://fatalfeel.blogspot.com/2013/12/faster-rcnn-in-pytorch.html

Refer to

• https://github.com/chenyuntc/simple-faster-rcnn-pytorch
• https://github.com/potterhsu/easy-faster-rcnn.pytorch