计算机工程

作者：于敏, 屈丹, 司念文

单位：1. 郑州大学 软件学院, 郑州 450000;2. 战略支援部队信息工程大学 信息系统工程学院, 郑州 450000

关键词：深度学习,目标检测,多光谱通道注意力,多尺度特征融合,完全交并比

基金：国家自然科学基金（62171470，61673395）。

针对经典一阶段目标检测算法RetinaNet难以充分提取不同阶段特征、边界框回归不够准确等问题，提出一个面向目标检测的改进型RetinaNet算法。在特征提取模块中加入多光谱通道注意力，将输入特征中的频率分量合并到注意力处理中，从而捕获特征原有的丰富信息。将多尺度特征融合模块添加到特征提取模块，多尺度特征融合模块包括1个路径聚合模块和1个特征融合操作，路径聚合模块通过搭建自底向上的路径，利用较浅特征层上精确的定位信号增强整个特征金字塔的信息流，特征融合操作通过融合来自每个阶段的特征信息优化多阶段特征的融合效果。此外，在边界框回归过程中引入完全交并比损失函数，从边界框的重叠面积、中心点距离和长宽比这3个重要的几何因素出发，提升回归过程的收敛速度与准确性。在MS COCO数据集和PASCAL VOC数据集上的实验结果表明，与RetinaNet算法相比，改进型RetinaNet算法在2个数据集上的平均精度分别提高了2.1、1.1个百分点，尤其对于MS COCO数据集中较大目标的检测，检测精度的提升效果更加显著。

来源：2022年第8期

《计算机工程》期刊编辑部

查看计算机工程杂志2022年第8期