Faster R-CNN: 基于区域提议网络的实时目标检测

Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks

罗斯·吉尔希克 Ross Girshick · Microsoft · 2015-06-04 · arXiv:1506.01497 ↗ · 被引 73697

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

最先进的目标检测网络依赖于区域提议算法来假设目标位置。SPPnet 和 Fast R-CNN 等进展减少了这些检测网络的运行时间,但区域提议计算成为瓶颈。在这项工作中,我们引入了一个区域提议网络(RPN),它与检测网络共享全图像卷积特征,从而实现了几乎无成本的区域提议。RPN 是一个全卷积网络,同时预测每个位置的目标边界和目标性得分。RPN 经过端到端训练以生成高质量的区域提议,供 Fast R-CNN 用于检测。我们进一步将 RPN 和 Fast R-CNN 合并为一个单一网络,通过共享它们的卷积特征——使用最近流行的神经网络“注意力”机制术语,RPN 组件告诉统一网络在哪里看。对于非常深的 VGG-16 模型,我们的检测系统在 GPU 上达到 5fps 的帧率(包括所有步骤),同时在 PASCAL VOC 2007、2012 和 MS COCO 数据集上仅使用每张图像 300 个提议就实现了最先进的目标检测精度。在 ILSVRC 和 COCO 2015 竞赛中,Faster R-CNN 和 RPN 是多个赛道第一名获奖作品的基础。代码已公开。

State-of-the-art object detection networks depend on region proposal algorithms to hypothesize object locations. Advances like SPPnet and Fast R-CNN have reduced the running time of these detection networks, exposing region proposal computation as a bottleneck. In this work, we introduce a Region Proposal Network (RPN) that shares full-image convolutional features with the detection network, thus enabling nearly cost-free region proposals. An RPN is a fully convolutional network that simultaneously predicts object bounds and objectness scores at each position. The RPN is trained end-to-end to generate high-quality region proposals, which are used by Fast R-CNN for detection. We further merge RPN and Fast R-CNN into a single network by sharing their convolutional features---using the recently popular terminology of neural networks with 'attention' mechanisms, the RPN component tells the unified network where to look. For the very deep VGG-16 model, our detection system has a frame rate of 5fps (including all steps) on a GPU, while achieving state-of-the-art object detection accuracy on PASCAL VOC 2007, 2012, and MS COCO datasets with only 300 proposals per image. In ILSVRC and COCO 2015 competitions, Faster R-CNN and RPN are the foundations of the 1st-place winning entries in several tracks. Code has been made publicly available.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 11)

阅读逐段中英对照全文 →