基于混合注意力和动态采样的遥感图像目标检测
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:


Remote Sensing Image Target Detection Based on Hybrid Attention and Dynamic Sampling
Author:
Affiliation:

Fund Project:

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

    针对遥感图像目标检测中的复杂背景干扰, 目标密集堆积导致遥感图像目标检测模型效果差的问题, 通过对YOLOv5s目标检测模型进行改进. 首先使用混合注意力机制对CBAM (convolutional block attention module)加以改进并添加至骨干网络中, 使得模型所提取到的特征含有局部信息和全局信息, 以加强模型识别复杂背景下目标的能力; 其次使用超轻量级的动态上采样器DySample结构, 减少模型参数并提高检测效率, 最后使用EIoU损失函数提高对于待检测目标的定位水平. 在RSOD和DIOR数据集上进行了实验验证, 结果表明, 改进后的YOLOv5s在遥感图像目标检测中的准确度要比原始模型高7.8%, 同时能够满足遥感图像实时目标检测的需求; 此外与其他目标检测模型相比, 改进模型也能保有一定优势.

    Abstract:

    Ineffective object recognition models occur in remote sensing images through complex background interference and dense target integration. To this end, this study improves the YOLOv5s object output model. First, a mixed attention menu is utilized to improve the convolutional attention model (CBAM) and add it to backbone networks. Accordingly, the extracted features of the model contain local and global information to enhance the model’s ability to identify targets in complex backgrounds. Then the study uses the ultra-light sampler DySample to reduce model parameters and improve model performance. Finally, the study employs the EIoU loss function to improve the positioning level of the target to be detected. Experimental verification of RSOD and DIOR data sets shows that the improved YOLOv5s has a 7.8% higher accuracy than the original model in detecting targets in remote sensing images, meeting the real-time detection requirements of targets in remote sensing images. In addition, the improved model retains the advantages it has in comparison to other object recognition models.

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

蔡清,王净雨,梁宏涛.基于混合注意力和动态采样的遥感图像目标检测.计算机系统应用,,():1-9

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-09-04
  • 最后修改日期:2024-09-24
  • 录用日期:
  • 在线发布日期: 2025-01-21
  • 出版日期:
文章二维码
您是第位访问者
版权所有:中国科学院软件研究所 京ICP备05046678号-3
地址:北京海淀区中关村南四街4号 中科院软件园区 7号楼305房间,邮政编码:100190
电话:010-62661041 传真: Email:csa (a) iscas.ac.cn
技术支持:北京勤云科技发展有限公司

京公网安备 11040202500063号