[发明专利]基于改进的多尺度自适应特征融合的DC-YOLOv4算法在审
申请号: | 202110802064.3 | 申请日: | 2021-07-15 |
公开(公告)号: | CN113591621A | 公开(公告)日: | 2021-11-02 |
发明(设计)人: | 杨晓雅;万冬厚;赵文君;田月媛;邓淼磊;张德贤 | 申请(专利权)人: | 河南工业大学 |
主分类号: | G06K9/00 | 分类号: | G06K9/00;G06K9/46;G06K9/62 |
代理公司: | 暂无信息 | 代理人: | 暂无信息 |
地址: | 450001 河南省郑州市高新技*** | 国省代码: | 河南;41 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 基于 改进 尺度 自适应 特征 融合 dc yolov4 算法 | ||
1.针对YOLOv4算法中小目标检测出现的问题,本发明提出了一种基于改进的多尺度自适应特征融合算法,并将这种算法与YOLOv4模型结合,称为DC-YOLOv4,其可以有效地提升小目标检测精度;
(1)在PAFPN结构的基础上,引入跳跃连接对同层级特征融合更多信息,就是将处于同一层级的初始输入特征图和输出特征节点进行连接,在两者之间简单增加一条额外的传输边进行特征结合,在不增加计算量的同时融合更丰富的同级特征;
其次,可以观察到C1,C2层级的浅层特征图在多尺度特征融合中并没有被使用过,但是它们含有大量的底层特征可以提升小目标检测的性能;因此,考虑将C1,C2层级特征跨尺度传输到P3层级,由于C1层级特征图的分辨率较高,所包含的特征信息过于接近初始图像像素信息,所以只将C2层级的信息进行特征融合,增加小目标检测能力;最后考虑到C2,C3层级的特征与P3融合时参与贡献不同,使用归一化加权融合来学习不同尺度特征的权重;归一化加权融合方式如公式所示;
w是可学习的权重,其参入计算时可以表示为标量(每个特征向量的值),矢量(针对输入的通道数量)以及多维向量(针对图像的所有像素点),所以必须将其权重进行归一化,使其大小限制在范围为[0,1]的区域内,则是为了避免数值不稳定的一个初始化小数;
多尺度自适应特征融合结构如图1所示,对应层级的特征图计算方式如公式所示,可以看出P5out、P4td、P4out、P3td、P3out这种经过多层级特征图输入的节点都是使用了权重归一化操作来自适应的进行特征融合;
(2)使用轻量级上采样算子CARAFE来代替第一次自底向上路径中上采样操作,上采样模块和下采样模块的位置如图2;在跨尺度双向特征融合网络特征中有两个融合模块,第一个是基于特征金字塔的上采样融合模块,第二个是自底向上路径里的下采样融合模块;两个融合模块在DC-YOLOv4算法中的特征融合方式如图3所示;
本发明的上采样融合模块是通过使用上采样算子CARAFE来取代原来融合过程中的双线性插值上采样算子;在自顶向下特征融合过程中,首先将C5特征图进行1×1卷积得到过渡特征图T5,然后通过CARAFE上采样算子进行2倍采样率计算得到的特征图和经过1×1卷积的C4特征图进行Concat特征融合,然后送入下一层级T4,之后迭代的完成三个层级的特征图上采样;CARAFE上采样算子的加入可以使得不同尺度的特征图内部的上下文语义信息传递更充分,信息的缺失降低,可以有效地提升检测效果;下采样融合模块主要是使用权重归一化操作对输入的不同尺度特征图进行调整,保证其都可以充分参入信息的融合过程。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于河南工业大学,未经河南工业大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202110802064.3/1.html,转载请声明来源钻瓜专利网。
- 上一篇:真空回潮机增湿控制方法
- 下一篇:一种准确测定混凝土表面裂缝的装置及方法