This is a PyTorch implementation of Faster RCNN. This project is mainly based on py-faster-rcnn and TFFRCNN.

For details about R-CNN please refer to the paper Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks by Shaoqing Ren, Kaiming He, Ross Girshick, Jian Sun.

• forward pass for detecting
• using models trained by Tensorflow
• roi pooling layer implemented by python and pytorch
• roi pooling layer with C extensions for pytorch
• backward pass for training

1. Clone the Faster R-CNN repository

   git clone [email protected]:longcw/faster_rcnn_pytorch.git

2. Build the Cython modules for nms and the roi_pooling layer

   cd faster_rcnn_pytorch/faster_rcnn
   ./make.sh

3. Download the trained model VGGnet_fast_rcnn_iter_70000.h5 and set the model path in demo.py

4. Run demo python demo.py