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

摘要：

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

关键词：目标检测、深度学习、特征金字塔、小目标

所属期刊栏目：45

分类号：TP391.4(计算技术、计算机技术)

资助基金：湖南省教育厅科学技术研究项目;长沙理工大学青年教师成长计划项目

在线出版日期：2023-04-28（万方平台首次上网日期，不代表论文的发表时间）

页数：共9页

页码：734-742