0
  • 聊天消息
  • 系统消息
  • 评论与回复
登录后你可以
  • 下载海量资料
  • 学习在线课程
  • 观看技术视频
  • 写文章/发帖/加入社区
创作中心

完善资料让更多小伙伴认识你,还能领取20积分哦,立即完善>

3天内不再提示

GA-RPN:Region Proposal by Guided Anchoring 引导锚点的建议区域网络

jf_96884364 来源:代码的路 作者:代码的路 2023-01-12 09:32 次阅读

原文链接

论文地址:https://arxiv.org/pdf/1901.03278.pdf

代码地址:GitHub - open-mmlab/mmdetection: OpenMMLab Detection Toolbox and Benchmark

1.RPN

RPN即Region Proposal Network,是用RON来选择感兴趣区域的,即proposal extraction。例如,如果一个区域的p>0.5,则认为这个区域中可能是80个类别中的某一类,具体是哪一类现在还不清楚。到此为止,网络只需要把这些可能含有物体的区域选取出来就可以了,这些被选取出来的区域又叫做ROI(Region of Interests),即感兴趣的区域。当然RPN同时也会在feature map上框定这些ROI感兴趣区域的大致位置,即输出Bounding Box。

RPN详细介绍:https://mp.weixin.qq.com/s/VXgbJPVoZKjcaZjuNwgh-A

2.Guided Anchoring

通常用(x,y,w,h)来描述一个anchor,即中心点坐标和宽高。文章将anchor的分布用条件概率来表示,公式为:

p(x,y,w,h|I)=p(x,y|I)p(w,h|x,y,I)

两个条件概率的分布,代表给定图像特征之后anchor的 中心点概率分布 ,和给定图像特征和中心点之后的 形状概率分布 。这样看来,原来我们所获取anchor的方法就可以看成上述条件概率分布的一个特例,即p(x,y|I)是均匀分布而p(w,h|x,y,I)是冲激函数。

根据上面的公式,anchor的生成过程可以分解为两个步骤,anchor位置预测和形状预测。

论文中用到的方法如下:

这个框架就是在原始的RPN的特征图基础上,采用两个分值分别预测anchor的位置和形状,然后再结合到一起得到anchor。之后采用一个Feature Adaption模块进行anchor特征的调整,得到新的特征图供之后的预测使用(anchor的分类和回归)。整个方法可以端到端训练,而且相比之前只是增加了3个1×1 conv 和一个3×3 deformable conv,带来的模型参数量变化很小。

(1)位置预测

位置预测分支的目标是预测哪些区域应该作为中心点来生成anchor,也是一个二分类问题,但是不同于RPN的分类,我们并不是预测每个点是前景还是背景,而是预测是不是物体中心。

我们将整个feature map的区域分为物体中心区域、外围区域和忽略区域,大致思路就是将groundtruth 框的中心一小块对应在feature map上的区域标为物体中心区域,在训练的时候作为 正样本 ,其余区域按照离中心的距离标为忽略或者 负样本 。最后通过选择对应概率值高于预定阈值的位置来确定可能存在对象活动的区域。F1 对输入的特征图使用 1×1 的卷积,得到与 F1 相同分辨率的输出,N_L 得到输出的每个位置的值表示原图I上对应位置出现物体的可能性,也就是概率图,最后通过选择对应概率值高于预定阈值的位置来确定可能存在对象活动的区域。

通过位置预测,我们可以筛选出一小部分区域作为anchor的候选中心点位置,使得anchor数量大大降低。这样在最后我们就可以只针对有anchor的地方进行计算。

(2)形状预测

形状预测分支是目标是给定anchor中心点,预测最佳的长和宽,这是一个回归问题。

采用1×1的卷积网络 N_s 输入 F_1,输出与 F_1 尺寸相同的2通道的特征图,每个通道分别代表 dw 和 dh,表示每个位置可能的最好的 anchor 尺寸。虽然我们的预测目标是 w 和 h,但是直接预测这两个数字不稳定,因为范围很大,所以将空间近似 [0,1000] 映射到了 [-1,1] 中,公式为:

w=\\sigma \\times s \\times e^{dw},w=\\sigma \\times s \\times e^{dh}

其中 s 是步幅,σ 是经验因子,实验中取 σ=8。实验中产生 dw,dh 的双通道映射,通过这个方程实现了逐像素转换。文章中直接用 IOU 作为监督来学习 w 和 h。

对于 anchor 和 ground truth 匹配问题,传统 RPN 都是直接计算 anchor 和所有 ground truth 的 IOU,然后将anchor 匹配给 IOU 最大的那个 ground truth,但是现在由于我们的改进,anchor 的 w 和 h 都是不确定的,是一个需要预测的变量。文中将这个 anchor 和某个 ground truth 的 IOU 表示为:

vIOU(a_{wh},gt)=\\max_{w>0,h>0}IOU_{normal}(a_{wh},gt)

我们不可能把所有可能的 w 和 h 遍历一遍求 IOU 的最大值,文中采用了9组可能的 w 和 h 作为样本,近似效果已经足够。

到这里我们就可以生成 anchor 了。这时所生成的 anchor 就是稀疏而且每个位置不一样的。实验可得此时的平均 recall 已经超过普通的 RPN 了,仅仅是增加了两个 conv。

(3)特征精调模块

由于每个位置的形状不同,大的anchor对应较大感受野,小的anchor对应小的感受野。所以不能像之前基于anchor的方法那样直接对feature map进行卷积来预测,而是要对feature map进行feature adaptation。作者利用可变形卷积(deformable convolution)的思想,根据形状对各个位置单独进行转换。

方法就是把anchor的形状信息直接融入到特征图当中,得到新的特征图去适应每个位置anchor的形状。这里就利用了上述的3×3的可变形卷积进行对原始特征图的修正,可变形卷积的变化量是通过anchor的w和h经过一个1×1 conv得到的。

f'_i=N_t(f_i,w_i,h_i)

其中,fi 是第 i 个位置的特征,(wi, hi) 是对应的 anchor 形状。NT 通过 3×3 的变形卷积实现。首先通过形状预测分支预测偏移字段 offset field,然后对带偏移的原始 feature map 做变形卷积获得 adapted features。之后进一步做分类和 bounding box 回归。

通过这样的操作,达到了让 feature 的有效范围和 anchor 形状更加接近的目的,同一个 conv 的不同位置也可以代表不同形状大小的 anchor 了。

文中实验结果示例:

学习更多编程知识,请关注我的公众号:

[代码的路]

声明:本文内容及配图由入驻作者撰写或者入驻合作网站授权转载。文章观点仅代表作者本人,不代表电子发烧友网立场。文章及其配图仅供工程师学习之用,如有内容侵权或者其他违规问题,请联系本站处理。 举报投诉
  • 神经网络
    +关注

    关注

    42

    文章

    4574

    浏览量

    98759
  • 代码
    +关注

    关注

    30

    文章

    4556

    浏览量

    66784
收藏 人收藏

    评论

    相关推荐

    易灵思Ti180报错分析cannot connect to more than 4 different clocks per region on left and right

    LVDS时钟有lvds_fast_clk和lvds_slow_clk两个,两组就会有4个时钟在Region clock R13区域。   Emulated  MIPI RX Function LVDS
    的头像 发表于 11-19 16:27 646次阅读
    易灵思Ti180报错分析cannot connect to more than 4 different clocks per <b class='flag-5'>region</b> on left and right

    OpenCV初学者如何提取这些不规则的ROI区域

    ROI是英文Region Of Interest的三个首字母缩写,很多时候我们对图像的分析就是对图像特定ROI的分析与理解,对细胞与医疗图像来说,ROI提取正确才可以进行后续的分析、测量、计算密度
    的头像 发表于 10-31 14:51 548次阅读

    STM32L5的系统新架构

    支持安全启动,信任链• Boot Lock,可保证唯一启动入口• HDP,可隐藏的安全用户闪存区域• TZ赋予STM32L5对隔离的硬件支持• 从内核扩展到全片系统• AHB总线主设备:DMA
    发表于 09-11 08:13

    存储区域网络快速参考指南

    电子发烧友网站提供《存储区域网络快速参考指南.pdf》资料免费下载
    发表于 08-29 09:28 0次下载
    存储<b class='flag-5'>区域</b><b class='flag-5'>网络</b>快速参考指南

    存储区域网络产品快速参考指南

    电子发烧友网站提供《存储区域网络产品快速参考指南.pdf》资料免费下载
    发表于 08-28 09:50 0次下载
    存储<b class='flag-5'>区域</b><b class='flag-5'>网络</b>产品快速参考指南

    faster rcnn网络结构详解(四个切入点)

    提到RPN网络,就不能不说anchors。所谓anchors,实际上就是一组由rpn/generate_anchors.py生成的矩形。直接运行作者demo中的generate_anchors.py可以得到以下输出。
    发表于 08-07 10:28 623次阅读
    faster rcnn<b class='flag-5'>网络</b>结构详解(四个切入点)

    浅谈GA-EMS导弹防御和先进武器系统技术

    GA-EMS的游荡弹药使作战人员能够以精确,控制和杀伤力打击视线以外的地面目标。由于它们能够长时间“徘徊”在区域上空,这些武器系统使士兵能够更好地定位、监视和引导导弹到达预定目标。
    发表于 07-24 12:23 633次阅读
    浅谈<b class='flag-5'>GA</b>-EMS导弹防御和先进武器系统技术

    开源网络引导固件iPXE正式支持龙架构

    开源网络引导固件iPXE集成龙架构补丁,正式支持龙架构(LoongArch)。用户可以使用iPXE从网络引导基于龙架构的操作系统。
    的头像 发表于 07-19 10:12 379次阅读
    开源<b class='flag-5'>网络</b><b class='flag-5'>引导</b>固件iPXE正式支持龙架构

    Go 1.21的PGO正式GA,性能提升,更快更猛!

    在 Go 语言中,最初关于 PGO 的提案是建议向 Go GC 工具链增加对配置文件引导优化 (PGO) 的支持,以便工具链能根据运行时信息执行特定于应用程序和工作负载的优化。
    的头像 发表于 06-28 16:47 767次阅读
    Go 1.21的PGO正式<b class='flag-5'>GA</b>,性能提升,更快更猛!

    GA4xxx 数据表

    GA4xxx 数据表
    发表于 06-27 20:59 0次下载
    <b class='flag-5'>GA</b>4xxx 数据表

    如何使用CCC数字钥匙R3汽车的BLE端?

    我们用 kw45b41zevk 板敲响了数字钥匙汽车演示代码。我不知道如何使用 CCC 数字钥匙 R3 汽车的 BLE 端。请给我一些演示文件。
    发表于 06-08 07:04

    恒讯科技分析:区域网络的类型有哪些?

    区域网络的类型有哪些?分别有LAN、MAN和WAN这三种主要类型的网络,设计用于在它们覆盖的区域上运行。它们之间有一些相同点和不同点。主要区别之一是它们覆盖的地理
    的头像 发表于 06-06 17:27 635次阅读

    PyTorch教程-14.8。基于区域的 CNN (R-CNN)

    14.8。基于区域的 CNN (R-CNN)¶ Colab [火炬]在 Colab 中打开笔记本 Colab [mxnet] Open the notebook in Colab
    的头像 发表于 06-05 15:44 387次阅读
    PyTorch教程-14.8。基于<b class='flag-5'>区域</b>的 CNN (R-CNN)

    FPGA中Bank和Clock Region之前有什么关系?

    FPGA中的Bank和Clock Region有什么关系?
    的头像 发表于 05-15 09:32 673次阅读
    FPGA中Bank和Clock <b class='flag-5'>Region</b>之前有什么关系?

    GA4xxx 数据表

    GA4xxx 数据表
    发表于 05-09 19:38 0次下载
    <b class='flag-5'>GA</b>4xxx 数据表