This code extends py-faster-rcnn by adding:
- ResNet implementation.
- Online Hard Example Mining.
- Caltech Dataset train and test interface.
- 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
- The caltech dataset interface based on GBJim-py-faster-rcnn
The installation and useage are same as Faster R-CNN
- clone the repository and caffe submodule
- build lib and caffe
- train your model command like:
./experiments/scripts/caltech.sh 0 ResNet50 reasonable