改进特征金字塔的小目标深度学习模型

作者：黄星威; 陈曦; 张塑凡

来源：计算机工程与科学, 2022, 1-9.

• 目标检测
• 深度学习
• 特征金字塔
• 小目标

现有的目标检测模型常采用特征金子塔的多尺度特征融合来提升小目标检测性能。然而，在特征金字塔的浅层特征层由于大目标的存在会削弱模型对小目标的检测，侧向连接会丢失高层特征层的语义信息，针对以上问题，提出了I-FPN特征金字塔。在浅层特征层，它抹去大目标信息让模型更关注小目标。在高层特征层，使用残差特征增强模块减少信息损失。此外，模型使用数据增广技术提升鲁棒性。I-FPN特征金字塔使用Resnet为主干网络，在VEDAI小目标数据集、PASCAL VOC通用目标数据集上实验。实验表明，在不影响检测速度条件下，在VEDAI测试集上较原特征金字塔mAP指标提升了2.4%，在VOC测试集上mAP指标提升了0.5%。

• 出版日期2022-1-19
• 单位长沙理工大学