/PS-PreciseRoIPooling

Position sensitive PreciseRoIPooling without roi coordinates gradient backward

Primary LanguageCuda

Position sensitive PreciseRoIPooling

I implement Position sensitive PreciseRoIPooling (Caffe ver.) base on https://github.com/vacancy/PreciseRoIPooling

After training PS-PreciseRoIPooling based light-head rcnn model for 20000 iterations, I got a not bad result.

I do not have enough time making experiments on PASCAL VOC or MSCOCO, If you get better results than PSRoi-Align or PSRoi-Pooling, please let me know, thank you.

My version is Position sensitive PreciseRoIPooling with no roi coordinates gradient backward, the source code already implements PreciseRoIPooling with it. You can get more details from the source code and paper.

sample

Usage

optional PSPRROIPoolingParameter psprroi_pooling_param = 164;
message PSPRROIPoolingParameter {
  required float spatial_scale = 1; 
  required int32 output_dim = 2; // output channel number
  required int32 group_size = 3; // number of groups to encode position-sensitive score maps
}