无锚框的轻量级遥感图像目标检测算法

来源：南京信息工程大学学报·自然科学版杂志2024年第2期北京时间：

作者：张云佐，武存宇，郭威，赵宁

单位：张云佐,石家庄铁道大学 信息科学与技术学院,石家庄, 050043;石家庄铁道大学 河北省电磁环境效应与信息处理重点实验室,石家庄, 050043,武存宇,石家庄铁道大学 信息科学与技术学院,石家庄, 050043,郭威,石家庄铁道大学 信息科学与技术学院,石家庄, 050043,赵宁,石家庄铁道大学 管理学院,石家庄, 050043

摘要：现有遥感图像目标检测算法存在参数量大、检测速度慢和难以部署于移动设备的问题,为此,本文提出了一种无锚框的轻量级遥感图像目标检测算法.首先设计了DWS-Sandglass轻量化模块以降低模型体积,并改进模型激活函数,以确保检测精度.然后引入无参数注意力模块SimAM,使网络能够专注于更重要的特征信息.最后对无锚框算法的冗余通道进行剪枝操作以减少模型参数量,并通过微调回升精度.在HRSC2016数据集上的实验结果表明,与当前主流的无锚框检测算法相比,该算法在检测精度相当的情况下检测速度更快、模型体积更小,更适合在移动设备部署.

关键词：计算机应用;遥感目标检测;轻量级;模型剪枝

基金资助：国家自然科学基金(61702347,62027801);河北省自然科学基金(F2022210007,F2017210161);河北省高等学校科学技术研究项目 (ZD2022100,QN2017132);中央引导地方科技发展资金(226Z0501G)