深度学习之检测模型-Faster RCNN

最新推荐文章于 2024-08-24 11:17:37 发布

leo_whz

最新推荐文章于 2024-08-24 11:17:37 发布

阅读量9.9k

点赞数 5

分类专栏： object_detection 文章标签：深度学习

本文链接：https://blog.csdn.net/whz1861/article/details/78768398

版权

Faster R-CNN是一种基于深度学习的目标检测模型，通过Region Proposal Network（RPN）实现区域提名，解决了传统方法中的计算瓶颈。RPN与分类、回归网络共享特征，提高效率。文章介绍了RPN、Translation-Invariant Anchors、Multi-Scale Anchors的概念，并详细讨论了损失函数、训练方法以及在PASCAL VOC和COCO数据集上的实验结果，验证了Faster R-CNN在目标检测领域的优越性能。

摘要由CSDN通过智能技术生成

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 net

Faster R-CNN 抛弃了 Selectave Search, 引入了 RPN 网络，使用区域提名，使得分类，回归一起共用卷积特征，从而进一步加速了检测过程。其中Faster RCNN使用了Anchor的思想对在Feature map上进行预处理，产生2w左右的候选框，使得RPN的回归变成了回归到anchor的相对位置，使得网络更加稳定。

思想

首先，Fast RCNN检测网络的主要思想：

使用一个简化的 SPP 层— RoI Pooling 层，操作与 SPP 相似 2. 训练和测试不再分多步
- a. 不用存储中间特征结果，可使得梯度通过 RoI Pooling 层直接传播
- b. 分类与回归用 Multi-task 的方式一起训练
SVD :使用 SVD 分解全连接层的参数矩阵，压缩为两个规模小很多的全连接层

解决 R-CNN 和 SPP-Net 中 2000 左右候选框带来的重复计算问题。
然后，overfeat虽然是一个端到端的训练过程，最终的网络进行了分类和回归网络多任务，但在预测的时候，仍然采用的是sliding window的思想。计算复杂度同样很大，因此作者提出了一种RPN网络，共享特征提取，使得网络的效率大大提升。

Region Proposal Networks

基础网络采用ZF或者VGG，则feature层的深度为256-d和512-d
然后将feature作为reg和cls的基础特征进行预测

数字	解释
171	ZF模型在feature层的感受野
228	VGG模型在feature层的感受野
256-d	ZF模型feature层的深度
512-d	VGG模型feature层的深度
3x3/1x1	feature层后接nxn(n=3)卷积进行特征提取，后面接两个1x1卷积分别对应reg和cls层
k	anchor的个数，论文 $k=9$ ，由3个scale和3个apect ratios组层
4k	对应RPN回归reg层输出的深度：[( $x_1,y_1,w_1,h_1$ ),( $x_2,y_2,w_2,h_2$ ) … ( $x_{k},y_{k},w_k,y_k$ )]
2k	对应RPN分类cls层输出的深度：[( $bg_1,fg_1$ ),( $bg_2,fg_2$ ) … ( b