Skip to content

Latest commit

 

History

History
 
 

cascade_rpn

Cascade RPN

Cascade RPN: Delving into High-Quality Region Proposal Network with Adaptive Convolution

Abstract

This paper considers an architecture referred to as Cascade Region Proposal Network (Cascade RPN) for improving the region-proposal quality and detection performance by systematically addressing the limitation of the conventional RPN that heuristically defines the anchors and aligns the features to the anchors. First, instead of using multiple anchors with predefined scales and aspect ratios, Cascade RPN relies on a single anchor per location and performs multi-stage refinement. Each stage is progressively more stringent in defining positive samples by starting out with an anchor-free metric followed by anchor-based metrics in the ensuing stages. Second, to attain alignment between the features and the anchors throughout the stages, adaptive convolution is proposed that takes the anchors in addition to the image features as its input and learns the sampled features guided by the anchors. A simple implementation of a two-stage Cascade RPN achieves AR 13.4 points higher than that of the conventional RPN, surpassing any existing region proposal methods. When adopting to Fast R-CNN and Faster R-CNN, Cascade RPN can improve the detection mAP by 3.1 and 3.5 points, respectively.

Results and Models

Region proposal performance

Method Backbone Style Mem (GB) Train time (s/iter) Inf time (fps) AR 1000 Config Download
CRPN R-50-FPN caffe - - - 72.0 config model

Detection performance

Method Proposal Backbone Style Schedule Mem (GB) Train time (s/iter) Inf time (fps) box AP Config Download
Fast R-CNN Cascade RPN R-50-FPN caffe 1x - - - 39.9 config model
Faster R-CNN Cascade RPN R-50-FPN caffe 1x - - - 40.4 config model

Citation

We provide the code for reproducing experiment results of Cascade RPN.

@inproceedings{vu2019cascade,
  title={Cascade RPN: Delving into High-Quality Region Proposal Network with Adaptive Convolution},
  author={Vu, Thang and Jang, Hyunjun and Pham, Trung X and Yoo, Chang D},
  booktitle={Conference on Neural Information Processing Systems (NeurIPS)},
  year={2019}
}