无锚框的轻量级遥感图像目标检测算法OA北大核心CSTPCD
现有遥感图像目标检测算法存在参数量大、检测速度慢和难以部署于移动设备的问题,为此,本文提出了一种无锚框的轻量级遥感图像目标检测算法.首先设计了DWS-Sandglass轻量化模块以降低模型体积,并改进模型激活函数,以确保检测精度.然后引入无参数注意力模块SimAM,使网络能够专注于更重要的特征信息.最后对无锚框算法的冗余通道进行剪枝操作以减少模型参数量,并通过微调回升精度.在HRSC2016数据集上的实验结果表明,与当前主流的无锚框检测算法相比,该算法在检测精度相当的情况下检测速度更快、模型体积更小,更适合在移动设备部署.
张云佐;武存宇;郭威;赵宁;
石家庄铁道大学信息科学与技术学院,石家庄050043 石家庄铁道大学河北省电磁环境效应与信息处理重点实验室,石家庄050043石家庄铁道大学信息科学与技术学院,石家庄050043石家庄铁道大学管理学院,石家庄050043
计算机与自动化
计算机应用遥感目标检测轻量级模型剪枝
《南京信息工程大学学报》 2024 (002)
P.212-220 / 9
国家自然科学基金(61702347,62027801);河北省自然科学基金(F2022210007,F2017210161);河北省高等学校科学技术研究项目(ZD2022100,QN2017132);中央引导地方科技发展资金(226Z0501G)。
评论