faster-rcnn-resnet

by Eniac-Xie

ResNet Implementation for Faster-rcnn

203 Stars 118 Forks Last release: Not found MIT License 19 Commits 0 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

Faster-RCNN-ResNet

This code extends py-faster-rcnn by adding ResNet implementation and Online Hard Example Mining.

This is a ResNet Implementation for Faster-RCNN. The faster rcnn code is based on py-faster-rcnn. The ohem code is based on ohem. To reduce the memory usage, we use batchnorm layer in Microsoft's caffe

Modification

  1. The caffe-fast-rcnn we use is a little different from the one py-faster-rcnn use, it uses the batchnorm layer from Microsoft's caffe to reduce the memory usage.
  2. Using the in-place eltwise sum within the PR
  3. To reduce the memory usage, we also release a pretrained ResNet-101 model in which batchnorm layer's parameters is merged into scale layer's, see tools/mergebnscale.py form more detail.
  4. Use Online-Hard-Example-Mining while training.

Installation

The usage is similar to py-faster-rcnn.

  1. Clone this repository

    Shell
    git clone https://github.com/Eniac-Xie/faster-rcnn-resnet.git
    
    We'll call the directory that you cloned faster-rcnn-resnet
    ROOT
  2. Clone the modified caffe-fast-rcnn

  cd $ROOT/
  git clone https://github.com/Eniac-Xie/caffe-fast-rcnn.git
  1. Build Cython module
   cd $ROOT/lib/
   make
  1. Build Caffe
   cd $ROOT/caffe-fast-rcnn
   make all -j8
   make pycaffe

Result

| | training data | test data | ohem | [email protected] | |------------------------|:-------------------:|:---------------------:|:------:|:-------------:| |Faster-RCNN, ResNet-101 | VOC 07+12 trainval | VOC 07 test | False | 78.78% |
|Faster-RCNN, ResNet-101 | VOC 07+12 trainval | VOC 07 test | True | 79.44% |

Testing

Download faster-rcnn-resnet weights from:

faster-rcnn-resnet without ohem (BaiduYun)

faster-rcnn-resnet without ohem (OneDrive)

faster-rcnn-resnet with ohem (BaiduYun)

faster-rcnn-resnet with ohem (OneDrive)

then you can do as follow:

   cd $ROOT/
   sh experiments/scripts/train_resnet101_bn_scale_merged_0712_end2end.sh
   make

or

   cd $ROOT/
   sh experiments/scripts/train_resnet101_bn_scale_merged_0712_end2end_ohem.sh
   make

Training

Download resnet-101 pretrained model, note that we use a modified version in which batchnorm layer's parameters is merged into scale layer's, you can download the model from Baidu Yun or OneDrive

then you can do as follow:

Shell
   cd $ROOT/
   sh experiments/scripts/train_resnet101_bn_scale_merged_0712_end2end.sh
or
Shell
   cd $ROOT/
   sh experiments/scripts/train_resnet101_bn_scale_merged_0712_end2end_ohem.sh

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.