leowangzi / CascadeRCNN

pytorch-1.1 torchvision-0.3

CascadeRCNN

pytorch-1.1 torchvision-0.3

References

• https://github.com/guoruoqian/cascade-rcnn_Pytorch
• https://github.com/jwyang/faster-rcnn.pytorch/tree/pytorch-1.0

Prerequisites

• Python 2.7 or 3.6
• Pytorch 1.1 or higher
• CUDA 9.0 or higher
• tensorboardX

Preparation

First of all, clone the code

git clone https://github.com/leowangzi/CascadeRCNN.git
cd CascadeRCNN

Then, create a folder:

mkdir data

or

ln -s [source_data] data

What we are doing and going to do

• [TODO] Support pytorch-1.1 (master).
• [TODO] Support torchvision-0.3 (master).
• soft-nms

Benchmarking

I benchmark this code thoroughly on pascal voc2007 and 07+12. Below are the results: