融合注意力机制的多尺度目标检测算法

被引：112

作者：

鞠默然 ^{[1
,2
,3
,4
,5
]}

罗江宁 ^{[6
]}

王仲博 ^{[1
,2
,3
,4
,5
]}

罗海波 ^{[1
,2
,4
,5
]}

机构：

[1] 中国科学院沈阳自动化研究所

[2] 中国科学院机器人与智能制造创新研究院

[3] 中国科学院大学

[4] 中国科学院光电信息处理重点实验室

[5] 辽宁省图像理解与视觉计算重点实验室

[6] 麦吉尔大学

来源：

光学学报 | 2020年 / 40卷 / 13期

关键词：

机器视觉; 卷积神经网络; 特征融合; 注意力机制; YOLO V3;

D O I：

暂无

中图分类号：

TP391.41 []; TP183 [人工神经网络与计算];

学科分类号：

140502 [人工智能];

摘要：

基于串联(concat)操作的特征融合方法仅仅融合了相邻尺度的特征,并没有充分利用来自其他尺度的输出特征。并且,串联操作只是在通道维度上将不同尺度的特征连接,不能反映不同通道间特征的相关性和重要性。针对这些问题,提出了一种基于注意力机制的特征融合算法。该算法利用注意力机制来融合不同尺度的特征,通过对每个通道的特征进行权重分配来学习不同通道间特征的相关性。将基于注意力机制的特征融合算法与YOLO V3相结合,构建多尺度目标检测器,并利用Focal loss和GIOU loss来设计检测器的损失函数。在PASCAL VOC和KITTI数据集上对不同算法进行对比实验,实验结果表明,多尺度目标检测器具有更高的检测精度和较快的检测速度。

引用

页码：132 / 140

页数：9

共 4 条

[1]

基于改进SSD的交通大场景多目标检测 [J].