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

作者:张云佐; 武存宇; 郭威; 赵宁
来源:南京信息工程大学学报(自然科学版), 2023, 1-11.
DOI:10.13878/j.cnki.jnuist.20230829001

摘要

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

全文