torchvision 中faster rcnn实现详解

最新推荐文章于 2024-04-21 15:46:09 发布
最新推荐文章于 2024-04-21 15:46:09 发布
阅读量2.3k 收藏 11
点赞数
分类专栏： 目标检测 文章标签： 计算机视觉
原文链接：https://zhuanlan.zhihu.com/p/133467109
版权

文章转载自：https://zhuanlan.zhihu.com/p/133467109

1.Faster RCNN 整体思路概述

如图1.1所示，Faster RCNN 的整体框架按照功能区分，大致分为4个模块，分别是特征提取网络backbone模块、RPN模块、RoI and RoI pooling模块和RCNN模块。

img图 1.1 Faster RCNN 整体框架

  • Backbone模块：主要负责接收输入数据，并进行数据预处理和特征提取得到输入图像对应的feature maps，并传递给下一层。这部分论文中用的VGG16和ZF框架，后来又有人用Resnet。
  • RPN network模块：这一模块主要有两个功能，一方面要生成一组proposals（图像中可能是前景的区域坐标），并将其传递给RoI模块；另一方面要计算RPN网络的损失，用于更新网络的参数。
  • RoI模块：对proposals进行降采样，并按proposals的坐标提取出feature maps中的特征，并将其传入下一层。
  • RCNN network模块：这一模块主要有两个功能，一方面用多层全连接网络对RoI传入的特征进行分类和回归，以得到预测目标的位置和标签；另一方面计算RCNN的损失，用于更新网络的参数。

如图1.1所示绿色框中的部分仅在训练时存在，整个结构的核心部分主要集中在后面三个部分，下面将对其进行详细的梳理。

2.Backbone模块

这一模块主要是利用效果比较好的卷积网络结构进行特征提取，这里就不详细介绍了。以VGG16为例，如图2.1所示，选取VGG16的前五层，取第五层pooling之前的数据作为feature maps。这里假设输入数据预处理后的尺度是则feature maps的尺度是 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-3OrcHma4-1617026232678)(https://www.zhihu.com/equation?tex=%5B3%5Ctimes600%5Ctimes800%5D)] ，则feature maps的尺度是 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-HApu9R4D-1617026232680)(https://www.zhihu.com/equation?tex=%5B512%5Ctimes37%5Ctimes50%5D)]。特征图中第二个和第三个维度中的每一个点都可以看成是原图中不同区域经过backbone层之后提取出的特征。

img图 2.1 Backbone 的结构图

3. RPN network模块

如图3.1所示为RPN network的总体框架。

img图 3.1 RPN network 整体框架及数据流关系（绿色部分表示仅在训练时存在）

3.1 RPN网络的输入输出和任务

  • 输入：feature maps
  • 输出：proposals（bounding box 的位置坐标）
  • 任务：训练RPN网络；根据feature maps计算proposals

这里思考几个问题，如果按照常规的有监督学习思路对RPN网络进行组织训练，即以ground true bounding box（ground true bbox）的坐标进行回归，以其类别标签进行分类。对于分类问题问题不大，但是因为ground true bbox 的标签在原图上的随机性比较大，数据分布比较广，很难找到合适的网络使回归损失收敛。

因此作者想到了另一个方法，不直接输出box的真值，而是通过一定的映射关系（后面会讲）输出ground true bounding box相对于固定基bounding box坐标下的偏移值。这样训练起来网络要容易收敛许多。

上面说的基bounding box就是下面要讲的anchor。

注：在进行梳理之前我们先假设现在网络已经训练成功，即网络可以理想的预测出相对于基bounding box的偏移值并且生成较为准确的前景背景分类label。

3.2 Anchor 与 anchor的生成

Anchor简单来说就是在原图上的一些矩型框，但是这一系列矩形框与feature maps相关联。如图3.2所示，对于feature maps上的某一个点都会生成一定数量的anchor，论文中采用三种尺寸（128， 256， 512），以及三种长宽比（1:1， 1:2 ， 2:1），组合起来一共有九种情况，即特征图上的每个点都会生成九个anchor。在代码中用每个anchor的左上角和右下角坐标来表示其位置。以一个特征点的九个anchor为例，就是生成如下形式的box坐标矩阵，维度为[9, 4]。

[[ -84.  -40.   99.   55.]
 
[-176.  -88.  191.  103.]
 
[-360. -184.  375.  199.]
 
[ -56.  -56.   71.   71.]
 
[-120. -120.  135.  135.]
 
[-248. -248.  263.  263.]
 
[ -36.  -80.   51.   95.]
 
[ -80. -168.   95.  183.]
 
[-168. -344.  183.  359.]]

img图 3.2 feature maps中某一点的anchors

但是刚才得到的坐标点是以这九个anchor的几何中心作为坐标原点得到的box坐标，因此还需要求anchors在原图坐标系下的box坐标。

首先要将feature maps映射到原图上，因为在backbone网络中对原图进行降采样得到feature maps，采样步长 stride = 16，因此将feature maps的坐标点乘以16，便可以映射回原图尺度。然后分别以这[37, 50]个点为原点，生成所有的anchors坐标。Anchor generate模块便是完成这部分功能，如图3.3所示。

img图 3.3 Anchor generate 功能示意图

生成的总anchors数为 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-dfd6L0q0-1617026232695)(https://www.zhihu.com/equation?tex=37%5Ctimes50%5Ctimes9%3D16650)] 个。这有些类似于穷举的方法，16650个预选框，总能有一些框比较贴合实际物体所在区域。但问题是这些贴合实际的框应该与ground true bounding box有些差异，并不完全吻合。所以需要一定的修正，这就是接下来要说的问题。

3.3 RPN网络的预测值

这部分功能主要由图3. 1.a中的Head模块完成，这里需要注意的是RPN网络的预测值和输出不同。此模块共有两个输出结果：

  • Objectness：预测anchors是前景还是背景的可能性大小
  • Pre bbox delta：预测anchors与ground true bounding box 的偏移量。

Head模块接收feature maps ,首先利用 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-cxzXlXma-1617026232696)(https://www.zhihu.com/equation?tex=%5B3%5Ctimes3%5D)] 的卷积进行更深层的特征提取，然后利用两个 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-IwDoHx83-1617026232699)(https://www.zhihu.com/equation?tex=%5B1%5Ctimes1%5D)] 卷积分别实现分类网络和回归网络。

在物体检测中通常将有物体的位置称为前景，没有物体的位置称为背景。在RPN的分类网络中，只需要区分出前景背景信息即可，因此这是一个二分类问题，考虑到每个特征点有9个anchor，所以分类网络中的卷积网络有 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-6CJMX07s-1617026232704)(https://www.zhihu.com/equation?tex=%5B9%5Ctimes2%5D%3D18)] 个通道。同理回归网络要输出对anchor的修正，所以卷积层对应有 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-TETx5ZB9-1617026232705)(https://www.zhihu.com/equation?tex=%5B9%5Ctimes4%5D%3D36)] 个通道。

img图 3.1.a Head 和Box decode模块

3.4 修正值与anchors关系

考虑某一特定的anchor，其中心点的坐标是 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-dVjZPY6L-1617026232710)(https://www.zhihu.com/equation?tex=%28x%2C+y%29)] ，宽是 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-7oSWt4Rz-1617026232711)(https://www.zhihu.com/equation?tex=w)] ，高是 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-GQJJ3Ovt-1617026232712)(https://www.zhihu.com/equation?tex=h)] ，假设它对应的ground true bounding box坐标为 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-NKel0MLT-1617026232713)(https://www.zhihu.com/equation?tex=%28x_0%2C+y_0%29)] ，宽是 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-OaqubOL9-1617026232716)(https://www.zhihu.com/equation?tex=w_0)] ，高是 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-UeR3vUgJ-1617026232716)(https://www.zhihu.com/equation?tex=h_0)] ，则对应的偏移值计算公式为：

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-3BBBtt6u-1617026232719)(https://www.zhihu.com/equation?tex=%5Cbegin%7Balign%7D+t_x+%26%3D+%5Cfrac%7B%28x_0±+x%29%7D%7Bw%7D%5C%5Ct_y+%26%3D+%5Cfrac%7B%28y_0±+y%29%7D%7By%7D%5C%5C+t_w+%26%3D+log%5Cfrac%7Bw_0%7D%7Bw%7D%5C%5C+t_h+%26%3D+log%5Cfrac%7Bh_0%7D%7Bh%7D+%5Cend%7Balign%7D%5Ctag1)]

3.3中Head模块输出的便是所有anchors的偏差值，利用式(1)，可以实现RPN网络的偏差值与原值的相互转换。

3.5 解码与编码

通过3.4和3.3知道head模块输出的并不是预测bounding box的位置坐标，要求位置坐标还需要利用anchors的坐标进行转换。这里约定两个过程：

  • 解码：已知偏差坐标[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-n8EXN1Pf-1617026232721)(https://www.zhihu.com/equation?tex=%28t_x%2C+t_y%2C+t_w%2C+t_h%29)]和基（在RPN中可以认为是anchor）坐标 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-ECs2DwOI-1617026232726)(https://www.zhihu.com/equation?tex=%28x_1%2C+y_1%2C+x_2%2C+y_2%29)] ，求预测bounding box坐标 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-yiWmiBgY-1617026232729)(https://www.zhihu.com/equation?tex=%28x_1%5E%2C+%2Cy_1%5E%2C%2C+x_2%5E%2C+%2Cy_2%5E%2C%29)] 的过程。
  • 编码：已知ground true bbox坐标 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-1zzEiqzT-1617026232730)(https://www.zhihu.com/equation?tex=%28x_1%5E%2C+%2Cy_1%5E%2C%2C+x_2%5E%2C+%2Cy_2%5E%2C%29)] 和基（在RPN中可以认为是anchor）坐标 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-f99ylBwC-1617026232732)(https://www.zhihu.com/equation?tex=%28x_1%2C+y_1%2C+x_2%2C+y_2%29)] ，求偏差坐标 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-9MCopgvc-1617026232733)(https://www.zhihu.com/equation?tex=%28t_x%2C+t_y%2C+t_w%2C+t_h%29)] 的过程。

图3.1.a中的Box decode 模块完成解码过程，对Head模块传来的的pre bbox delta解码之后生成在原图坐标系下的pre bounding box 的位置坐标。

图3.1.c中Box encode 完成编码过程。

具体实现还要考虑到原始图像的大小，进而对解码结果进行修整。

3.6 NMS 与生成proposals

现在假设RPN模型已经训练成功，网络处于inference阶段，Head模块可以输出较理想的objectness 和 pre bbox delta 值。在经过box decode之后会输出所有anchors的objectness得分和pre bounding box坐标。最后利用filter proposals模块完成NMS和生成proposals的任务。如图3.1.b所示，具体描述如下：

  • 根据objectness的前景得分进行排序，按一定比例选出前12000个pre bbox 得到最初的建议区域。
  • 考虑到一个物体可能会有多个bounding box重叠对应，所以在应用非极大值抑制（NMS）将重叠框去掉。
  • 最后在剩余的bbox中，根据objectness前景得分随机抽取2000个box，作为最终的proposals。

img图3.1.b Filter proposalss模块

3.7 RPN network的训练

为了让RPN网络能够提出良好的proposals，需要对网络进行训练。就像传统的有监督网络一样，在训练前我们需要明确，模型的预测值、真值和损失函数这三个要素。预测值在3.3中已经介绍，这里介绍真值和损失函数。图3.1.c 为训练部分结构图。

img图 3.1.c RPN 训练部分结构图

3.7.1 RPN network的真值

RPN网络的预测值是对anchors的分类和回归修正，其真值也有两个——对应anchors的类别标签和真实偏差值。这一部分主要由Assign targets to anchors模块完成。

注意这两者的求取都是建立在anchors的基础之上。

为了便于理解，先介绍以下ground true的形式：

  • Ground true label：假设需要区分的类别有C种，则label的值属于整数[0, C]，label的个数为输入图片中目标的个数。假如C=3， 有2个目标，则label = [1, 0]或[0, 2]……
  • Ground true bounding box：形式为box的左上角和右下角坐标如 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-s21VWGti-1617026232737)(https://www.zhihu.com/equation?tex=%28x_1%2C+y_1%2C+x_2%2C+y_2%29)] ，box个数为输入图片中目标的个数。假如C=3， 有2个目标，则box{[1, 2, 3, 4], [5, 6, 7, 8]}。
  • 注：与box中对应位置的label值即为该位置的目标类别。

（1）真值标签（target labels）的求取

在这一部分标签主要要区分出前景和背景，不需要具体的分类，因此标签值有两个：

  • 0：背景
  • 1：前景

前景背景的区分是通过计算ground true bounding box与anchors的IoU值来区分的，这里需要计算每个ground true bbox和anchors的IoU，也就是说会形成一个IoU矩阵。假设ground true bbox有N个，则IoU大小为[N, 166500]具体的判断标准如下：

  • 对于任何一个anchor与所有的gt box最大IoU小于阈值0.3（可以调整的参数），则视为负样本。
  • 对于任何一个gt box，与其有最大IoU的anchor视为正样本。
  • 对于任何一个anchor，与所有gt box的IoU大于阈值0.7（可以调整的参数），则视为正样本。
  • -1标签：在编程时，可以将IoU值在0.3-0.7之间的anchor标签置位-1，作为无效anchor的标记。这部分anchor不参与分类和回归损失的计算。

（2）真实坐标（target regression）的求取

这部分的主要思路是，利用IoU矩阵，对于某一anchor找到与其IoU值最大的gt box坐标，作为其真实坐标，依次找到所有的anchors真值。

注：

1）这里有一个问题：对某些不符合前景标准的anchor也进行了真值赋值，但是在后续计算中，会根据target label进行抽样，不符合前景背景标准的target regression不会被计算到。

2）这里求得的所有anchor的真值坐标是 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-TP0Tvvlv-1617026232738)(https://www.zhihu.com/equation?tex=%28x_1%2C+y_1%2C+x_2%2C+y_2%29)] 形式的，还需要对其进行编码操作（box code）才能得到RPN网络的回归真值target regression。

3.7.2 RPN network损失函数的计算

经过3.7.1和3.2，RPN获得了网络的预测值和真值，现在可以计算损失函数了。

（1）抽取正负样本

由于网络预测的bounding box综数接近两万，并且大部分box标签都是背景，如果都计算损失的话则正负样本失去平衡，不利于网络收敛。因此，RPN默认选择256（可调整）个bounding box进行损失计算，其中最多不超过128（可调整）个正样本。如果超过则进行随机抽取。

（2）损失函数的计算

RPN损失函数主要有两部分组成，即分类损失和回归损失，如下所示：

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-CPWZyu5m-1617026232739)(https://www.zhihu.com/equation?tex=L%28%7Bp_i%2C+t_i%7D%29+%3D+%5Cfrac%7B1%7D%7BN_%7Bcls%7D%7D%5Csum_iL_%7Bcls%7D%28p_i%2C+p_i%5E%2A%29%2B%5Clambda%5Cfrac%7B1%7D%7BN_%7Breg%7D%7D%5Csum_ip_i%5E%2AL_%7Breg%7D%28t_i%2C+t_i%5E%2A%29%5Ctag%7B2%7D)]

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-Rf5fD5jC-1617026232740)(https://www.zhihu.com/equation?tex=L_%7Bcls%7D%28p_i%2C+p_i%5E%2A%29)] 代表了晒算出来的256个box的分类损失，这里是一个二分类问题，使用的是交叉熵函数。其中 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-s35kOkym-1617026232741)(https://www.zhihu.com/equation?tex=p_i)] 为真值， [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-xg4QnkS9-1617026232742)(https://www.zhihu.com/equation?tex=p_i%5E%2A)] 为预测值。

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-S8qO5sHb-1617026232743)(https://www.zhihu.com/equation?tex=L_%7Breg%7D%28t_i%2C+t_i%5E%2A%29)] 代表了回归损失，回归损失使用的是 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-MpWgQw1d-1617026232744)(https://www.zhihu.com/equation?tex=smooth_%7BL1%7D)] 函数，如下所示。其中 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-cwbHUaeq-1617026232745)(https://www.zhihu.com/equation?tex=t_i)] 为真值， [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-QzzsYUDi-1617026232747)(https://www.zhihu.com/equation?tex=t_i%5E%2A)] 为预测值。

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-HfW3K5n4-1617026232748)(https://www.zhihu.com/equation?tex=L_%7Breg%7D%28t_i%2C+t_i%5E%2A%29+%3D+%5Csum_%7Bi%5Cin%7Bx%2C+y%2C+w%2C+h%7D%7Dsmooth_%7BL1%7D%28t_i%2C+t_i%5E%2A%29%5C%5C+smooth_%7BL1%7D%28t_i%2C+t_i%5E%2A%29+%3D+++%5Cbegin%7Bcases%7D+0.5+%5Ctimes+x%5E2+%26%26if+%5C%2C%5C%2C%5C%2C%7Cx%7C%3C1%5C%5C+%7Cx%7C-0.5%26%26otherwise+%5Cend%7Bcases%7D+%5Ctag3)]

注：在训练阶段RPN的损失函数会传递到RCNN部分一起计算总的损失函数，进而进行反馈，改善网络效果。

至此RPN部分的所有内容梳理完毕。

附：RCNN 模块和 RPN 的对比分析

RCNN部分和RPN十分的相似，本部分希望通过对二者的差异分析，使读者对最后两模块的作用和结构有一个整体的认识。

RPN的核心思想可以说是：对真值与anchors的偏差进行预测。相对来说RCNN模块的核心思想可以说是：对真值与proposals的偏差进行预测。所以所二者在实现上十分相似，概括的说有“两区别、一相似”。

  • 区别一：feature maps不同

RPN 的feature maps是经过backbone提取的全局特征，而 RCNN 的pooling maps 是利用proposals的先验，以及原图和feature maps 的对应关系提取出的对应于每一个proposals的局部特征图。这部分提取的功能由 RoI and RoI pooling 部分完成。而且由于后续的网络是全连接网络，RoI and RoI pooling 模块还需要将提取后的局部特征图pooling到统一的尺寸。

  • 区别二：预测的网络结构预测类别数量不同

RPN网络用的是卷积网络，RCNN用的是全连接网络。

RPN网络是二分类问题，RCNN网络是C+1分类问题。

  • 相似一：给 proposals 分配真值的过程相似

RPN 中通过计算anchors和ground true bounding box的IoU矩阵来分配anchors对应的target labels 和 bounding box。

RCNN 中通过计算proposals和ground true bounding box的IoU矩阵来分配proposals的target labels 和bounding box。

4. RoI and RoI pooling模块

4.1 RoIs的获得

训练时由RPN网络传输过来的proposals的数量为2000个，其中仍然有很多背景框，真正的存在目标的前景占比很少。因此在进行特征提取之前要对proposals进行再次筛选。筛选的流程如图4.1所示。

  • 计算IoU矩阵和分配标签的过程和RPN网络中Assign targets to anchors模块一致，这里只是将anchors替换成了proposals。这里就不重复介绍了。
  • 在分配完标签之后，在所有正负样本中选出256（参数可调节）个样本，正样本数按照一定比例提取，同样比例可以调整。

最后选出的256个proposals即为RoIs。

img图 4.1 RoI 和 RoI pooling 结构图

4.2 RoI pooling

在4.1中得到的RoIs区域，实际是对应原图中可能是前景和背景的bbox坐标，我们需要将这个坐标映射到feature maps上，进而提取出RoI区域对应特征图。然后再传递给后面的全连接层进行分类和回归。

这里有一个问题是，特征最后要输入到全连接层进行计算，所以要求输入的特征图大小是一样的。而由于RoI 的bbox大小不同，其提取的特征大小也不同，因此特征传递给FC层之前需要进行一定的处理，这就是RoI pooling的主要作用。

论文中统一将特征图pooling到 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-0whYovBg-1617026232750)(https://www.zhihu.com/equation?tex=%5B7%5Ctimes7%5D)] 大小。

具体的方法这里就不详细介绍了。

注：由于RoI pooling在实现的时候会多次取整，导致精度不高。后来常用maskRCNN中的RoI Align代替。

5. RCNN network

5.1 RCNN的输入输出和任务

  • 输入：RoI and RoI pooling 网络传递过来的 固定维度的pooling maps特征图
  • 输出：目标类别和位置偏差的预测值，以及RCNN Loss。
  • 任务：训练RCNN网络；对pooling maps进行学习，利用三层全连接网络进行标签分类及位置回归。

5.2 全连接模块

这里用三层全连接层对特征进行处理，如图所示。前两层是VGG16中的第六第七层，最后一层是两个全连接层，分别对256个RoIs区域进行类别预测和位置预测。

img图 5.1 全连接网络结构图

图中C代表需要检测的类别，背景算作第0类，所以类别预测输出的维度是 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-UvfvNSCO-1617026232752)(https://www.zhihu.com/equation?tex=%5B256%5Ctimes%281%2BC%29%5D)] 。位置预测需要对每一类都预测出位置所以输出维度是 [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-U2CinFvH-1617026232753)(https://www.zhihu.com/equation?tex=%5B256%5Ctimes%281%2BC%29%5Ctimes4%5D)] 。

5.3 RCNN网络的训练

在训练之前我们来梳理以下RCNN网络的预测值和真值。RCNN 的预测值在5.2中已经求出，就是RCNN网络的输出值。

（1）RCNN网络的真值

RCNN网络的真值是RoIs区域的类别标签和与对应的真实坐标的偏差。RoIs区域对应的真实坐标在RoI模块中的assign targets中实际已经得到，具体过程同3.7.1中RPN部分一致。

注：这里得到的RoIs target location 是编码前的表示，所以需要进行box code计算偏差，才是真正的RCNN网络的真值。

（2）损失函数

损失函数的计算和RPN部分相同，不在重复。

  • 需要注意的是这里的分类损失是C+1类，而不是二分类；
  • 回归损失只计算正样本部分对应类别的损失。

举例说明第二点。假设第3个区域为正样本且是第K类，则对应pre bbox delta中的[3，K, ：]的delta值参与损失计算。

至此RCNN中关于训练和inference的思路全部梳理完毕。图5.2展示了RoI and RoI pooling和RCNN模块之间的关系和数据流。

img图5.2 RoI and RoI pooling及RCNN模块之间的关系和数据流（绿色部分表示仅在训练时存在）

6. 总结

本文主要从编程实现的角度梳理了Faster RCNN的流程，重点介绍了RPN部分的实现。RoI and RoI pooling和RCNN部分的结构和RPN网络实际上十分类似，很多的思路都是想通的，因此介绍的并不详细。

比如：RPN的预测值是对anchors进行修正，得到proposals；RCNN网络其实是对RoI进行修正得到最终的预测结果。其修正的目的都是为了使被修正的bbox经过修正后能够更接近ground true bbox。

关于RoI pooling和RoI Align：其二者的具体方法本文没有介绍，这部分的内容并不影响读者对整个网络的理解，如果想要了解，建议阅读原文、源码或其他博客。

关于参数上细节：整个框架中有许多可调节的超参数，如anchor的大小 和 长宽比、生成proposals时的下采样数等等。而且有些参数在训练和inference时并不相同。如果读者想要了解请参考源码。

关于参考代码：本文的主要参考代码是torchvision包中自带的fasterrcnn_resnet50_fpn()实现。

关于配图：图3.3中的部分图直接截取自

Object Detection and Classification using R-CNNswww.telesens.co

附：补充一个整体结构图

using R-CNNs](https://link.zhihu.com/?target=http%3A//www.telesens.co/2018/03/11/object-detection-and-classification-using-r-cnns/)www.telesens.co

附：补充一个整体结构图

img

JaciusCV
关注
专栏目录
PyTorch深度学习实战（21）——从零开始实现Faster R-CNN目标检测
盼小辉丶的博客
10-16 4266
Faster R-CNN 提出了一种全新的目标检测框架，将候选框生成和目标分类合并到一个网络中，实现了端到端的训练，可以同时优化候选框生成和目标分类任务，提高了检测的准确性和效率。同时，利用共享的卷积特征可以加速特征提取的计算，进一步提高了检测速度。在本节中，将介绍 Faster R-CNN 的工作原理，然后在自定义数据集上训练 Faster R-CNN 目标检测模型。
pytorch深度学习入门（9）之-使用TORCHVISION进行目标检测
ajunbin859的专栏
11-14 2833
用于训练对象检测、实例分割和人物关键点检测的参考脚本可以轻松支持添加新的自定义数据集。数据集应继承自标准 torch.utils.data.Dataset类，并实现__len__和getitem。我们需要的唯一特殊性是数据集__getitem__ 应返回一个元组：image：torchvision.tv_tensors.Imageshape 、纯张量或 size 的 PIL 图像[3, H, W](H, W)target：包含以下字段的字典。
python 利用torchvision 搭建fasterRcnn
qq_38878964的博客
09-20 909
python 利用torchvision 搭建fasterRcnn
Faster-RCNN-with-torchvision:使用Torchvision重新实现Faster-rcnn
05-04
Faster-RCNN-with-torchvision 通过pytorch torchvision官方提供的模块构建Faster-RCNN，供检测及学习使用。 代码博客解析地址： 模型地址： 安装 代码及环境搭建 $ git clone https://github.com/supernotman/Faster-RCNN-with-torchvision.git $ cd Faster-RCNN-with-torchvision/ $ sudo pip install -r requirements.txt 数据准备 目前代码仅支持coco数据集的训练 下载coco2017数据集 下载后数据集文件结构如下： coco/ 2017/ annotations/ test2017/ train2017/ val2017/ 训练及测试
faster R-CNN 模型解析（torchvision版）
孙永杰的博客
01-06 2969
torchvision中已经为我们实现好了faster R-CNN模型，我们只需要调用即可。本文将对该模型做进一步的分析，使自己能够在使用过程中更加得心应手。
利用Pytorch torchvision完成Faster-rcnn目标检测demo及源码详解
热门推荐
watermelon1123的博客
08-27 2万+
Torchvision更新到0.3.0后支持了更多的功能，其中新增模块detection中实现了整个faster-rcnn的功能。本博客主要讲述如何通过torchvision和pytorch使用faster-rcnn，并提供一个demo和对应代码及解析注释。 目录 如果你不想深入了解原理和训练，只想用Faster-rcnn做目标检测任务的demo，请看这里 torchvision中Faste...
0005-Pytorch入门-使用torchvision自带的faster-rcnn进行目标检测
左手coding，右手cooking
07-30 1515
# -*- coding: utf-8 -*- """ Created on Thu Jul 30 08:47:12 2020 @author: Johnson """ import cv2 as cv import torchvision import torch from torchvision import transforms import numpy as np with open("coco.names") as f: #获取类别名称 coco_names = [line.str
Pytorch torchvision完成Faster-rcnn目标检测demo及源码详解
sinat_28371057的博客
03-10 2134
Torchvision更新到0.3.0后支持了更多的功能，其中新增模块detection中实现了整个faster-rcnn的功能。本博客主要讲述如何通过torchvision和pytorch使用faster-rcnn，并提供一个demo和对应代码及解析注释。 目录 如果你不想深入了解原理和训练，只想用Faster-rcnn做目标检测，请看这里 torchvision中Faster-rcnn接口 一个demo 使用方法 如果你想深入了解原理，并训练自己的模型 环境搭建 准备训练数据 模型训练
faster rcnn代码详解pytorch
03-16
在PyTorch中实现Faster RCNN，可以使用torchvision中的models和transforms模块，以及torch.utils.data中的DataLoader和Dataset模块。具体实现步骤如下： 1. 加载数据集 使用torchvision中的transforms模块对数据...
【目标检测系列：三】Faster RCNN simple-faster-rcnn-pytorch 实现详解
鹿鹿的博客
06-20 3731
Pytorch https://github.com/jwyang/faster-rcnn.pytorch https://github.com/ruotianluo/pytorch-faster-rcnn https://github.com/chenyuntc/simple-faster-rcnn-pytorch 跑了这个 ☺ 1. Prepare data Pascal VOC2007 D...
【PyTorch实战演练】Faster R-CNN介绍以及通过预训练模型30行代码实现目标检测_faster rcnn预训练模型
最新发布
2401_84520150的博客
04-21 927
按照国际惯例，首先声明：本文只是我自己学习的理解，虽然参考了他人的宝贵见解及成果，但是内容可能存在不准确的地方。如果发现文中错误，希望批评指正，共同进步。本文介绍Faster R-CNN的结构及原理，并基于PyTorch官方文档使用预训练好的模型进行目标检测实例。关于面试刷题也是有方法可言的，建议最好是按照专题来进行，然后由基础到高级，由浅入深来，效果会更好。Java基础部分算法与编程数据库部分流行的框架与新技术（Spring+SpringCloud+SpringCloudAlibaba）
Python-用PyTorch实现FasterRCNN
08-09
用PyTorch实现Faster RCNN
TorchVision中使用FasterRCNN+ResNet50+FPN进行目标检测
网络资源是无限的
11-19 1万+
TorchVision中给出了使用ResNet-50-FPN主干(backbone)构建Faster R-CNN的pretrained模型，模型存放位置为https://download.pytorch.org/models/fasterrcnn_resnet50_fpn_coco-258fb6c6.pth，可通过fasterrcnn_resnet50_fpn函数下载，此函数实现在torchvison/models/detection/faster_rcnn.py中，下载后在Ubuntu上存放在...
Pytorch torchvision构建Faster-rcnn（二）----基础网络
watermelon1123的博客
08-19 8421
torchvision中提供了通过Resnet + FPN的方式构建基础网络，这里以resnet50为例实现基础网络的构建。 目录 BackboneWithFPN IntermediateLayerGetter FeaturePyramidNetwork BackboneWithFPN 类BackboneWithFPN的在backbon_utils.py中，功能是给骨干网络加上FPN，...
torchvision Faster-RCNN ResNet-50 FPN代码解析（RPN）
大飞的博客
10-08 3282
总体架构 RPNHead
torchvision源码解读——Faster-RCNN
graviton的博客
11-26 195
概述 torchvision是用pytorch写成的著名的计算机视觉库，这篇博客介绍经典的目标检测算法Faster-RCNN，而torchvision中的Faster-RCNN选取的特征提取网络是ResNet-50,并且还加上了后来的研究成果FPN(特征金子塔) ...
Faster_R_CNN源码详细解析
a535652512的博客
05-06 594
Faster R-CNN整体架构 首先使用共享卷积层为全图提取特征feature maps 将得到的feature maps送入RPN，RPN会产生接近两千个候选框proposals RoI Pooling Layer根据RPN的输出在feature map上面找到proposals对应的对应区域也就是ROI（region of interest），对不同大小ROI输出固定大小的feature...
Faster RCNN pytroch代码
09-26
在提供的`faster_rcnn_pytorch-master`压缩包中，应包含了完整的Faster R-CNN PyTorch实现代码，包括数据预处理、模型定义、训练和推理的脚本。初学者可以通过阅读和运行这些代码，深入理解Faster R-CNN的工作原理和...
深度学习和目标检测系列教程 10-300：通过torch训练第一个Faster-RCNN模型
微信号：RunsenLiu
07-07 2002
@Author：Runsen 上次介绍了Faster-RCNN模型，那么今天就开始训练第一个Faster-RCNN模型。
Pytorch torchvision构建Faster-rcnn（三）----Region Proposal Network
watermelon1123的博客
08-20 6791
RPN是two-stage的标志性结构，并且其本身也是一个二分类的目标检测网络，因此在faster-rcnn的整个网络结构中能看到anchor的使用，回归和分类等操作，这里讲具体介绍一下。 整个rpn部分代码在torchvison/models/detection/rpn.py中，其中定义了RPNHead，AnchorGenerator，RegionProposalNetwork三个模块。 目...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值