this respository is forked from [cafffe_faster_rcnn_renet]( https://github.com/Enia c-Xie/caffe-fast-rcnn.git), thanks for Eniac-Xie
Now I want to do:
implement FPN(Feature Pyramid Network)
Test on VOC
Test on COCO
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
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/merge_bn_scale.py form more detail.
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:
cd $ROOT/
sh experiments/scripts/train_resnet101_bn_scale_merged_0712_end2end.sh
or
cd $ROOT/
sh experiments/scripts/train_resnet101_bn_scale_merged_0712_end2end_ohem.sh