pytorch-1.1 torchvision-0.3
- https://github.com/guoruoqian/cascade-rcnn_Pytorch
- https://github.com/jwyang/faster-rcnn.pytorch/tree/pytorch-1.0
- Python 2.7 or 3.6
- Pytorch 1.1 or higher
- CUDA 9.0 or higher
- tensorboardX
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
- [TODO] Support pytorch-1.1 (master).
- [TODO] Support torchvision-0.3 (master).
- soft-nms
I benchmark this code thoroughly on pascal voc2007 and 07+12. Below are the results: