无锚框的轻量级遥感图像目标检测算法
DOI:
作者:
作者单位:

1.石家庄铁道大学;2.石家庄铁道大学 管理学院

作者简介:

通讯作者:

中图分类号:

基金项目:

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


Lightweight remote sensing image target detection algorithm without anchor frame
Author:
Affiliation:

1.Shijiazhuang Tiedao University;2.School of Management, Shijiazhuang Tiedao University

Fund Project:

the National Natural Science Foundation of China (No. 61702347, No. 62027801), the Natural Science Foundation of Hebei Province (No. F2022210007, No. F2017210161), the Science and Technology Project of Hebei Education Department (No. ZD2022100, No. QN2017132), the Central Guidance on Local Science and Technology Development Fund (No. 226Z0501G).

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

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

    Abstract:

    The existing remote sensing image object detection algorithms have the problem of large parameter quantities and difficulty in deploying on mobile devices. Therefore, this paper proposes a lightweight remote sensing image object detection algorithm without anchor boxes. Firstly, the DWS-Sandglass lightweight module was designed to reduce the model volume, and then the activation function of the model was improved to ensure the detection accuracy. Then, a parameter free attention module SimAM is introduced to enable the network to focus on more important feature information. Finally, prune the redundant channels of the anchor free box algorithm to reduce the number of model parameters, and fine tune to improve accuracy. The experimental results on the HRSC2016 dataset show that compared with the current mainstream anchor free frame detection algorithms, this algorithm has faster detection speed and smaller model size, making it more suitable for deployment on mobile devices with comparable detection accuracy.

    参考文献
    相似文献
    引证文献
引用本文

张云佐,武存宇,郭威,赵宁.无锚框的轻量级遥感图像目标检测算法[J].南京信息工程大学学报,,():

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-08-29
  • 最后修改日期:2023-10-14
  • 录用日期:2023-10-16
  • 在线发布日期:
  • 出版日期:

地址:江苏省南京市宁六路219号    邮编:210044

联系电话:025-58731025    E-mail:nxdxb@nuist.edu.cn

南京信息工程大学学报 ® 2024 版权所有  技术支持:北京勤云科技发展有限公司