期刊文献+

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

Lightweight remote sensing image target detection without anchor frame
下载PDF
导出
摘要 现有遥感图像目标检测算法存在参数量大、检测速度慢和难以部署于移动设备的问题,为此,本文提出了一种无锚框的轻量级遥感图像目标检测算法.首先设计了DWS-Sandglass轻量化模块以降低模型体积,并改进模型激活函数,以确保检测精度.然后引入无参数注意力模块SimAM,使网络能够专注于更重要的特征信息.最后对无锚框算法的冗余通道进行剪枝操作以减少模型参数量,并通过微调回升精度.在HRSC2016数据集上的实验结果表明,与当前主流的无锚框检测算法相比,该算法在检测精度相当的情况下检测速度更快、模型体积更小,更适合在移动设备部署. The existing remote sensing image object detection algorithms have been frustrated by large parameter quantities,slow detection speed and inability to deploy on mobile devices.Here,we propose a lightweight remote sensing image object detection algorithm without anchor frames.First,a DWS-Sandglass lightweight module is designed to reduce the model volume,and the activation function of the model is improved to ensure detection accuracy.Then,a parameter free attention module SimAM is introduced to enable the network to focus on more important feature information.Finally,the redundant channels of the anchor frame free algorithm are pruned to reduce the number of model parameters,and the accuracy is improved by fine tuning.The experimental results on HRSC2016 dataset show that compared with current mainstream detection algorithms free of anchor frame,the proposed algorithm has faster detection speed and smaller model size,making it more suitable for deployment on mobile devices with comparable detection accuracy.
作者 张云佐 武存宇 郭威 赵宁 ZHANG Yunzuo;WU Cunyu;GUO Wei;ZHAO Ning(School of Information Science and Technology,Shijiazhuang Tiedao University,Shijiazhuang 050043,China;Hebei Key Laboratory of Electromagnetic Environmental Effects and Information Processing,Shijiazhuang Tiedao University,Shijiazhuang 050043,China;School of Management,Shijiazhuang Tiedao University,Shijiazhuang 050043,China)
出处 《南京信息工程大学学报》 CAS 北大核心 2024年第2期212-220,共9页 Journal of Nanjing University of Information Science & Technology(Natural Science Edition)
基金 国家自然科学基金(61702347,62027801) 河北省自然科学基金(F2022210007,F2017210161) 河北省高等学校科学技术研究项目(ZD2022100,QN2017132) 中央引导地方科技发展资金(226Z0501G)。
关键词 计算机应用 遥感目标检测 轻量级 模型剪枝 computer applications remote sensing target detection lightweight model pruning
  • 引文网络
  • 相关文献

参考文献6

二级参考文献40

共引文献36

;
使用帮助 返回顶部