改进YOLOv5s遥感图像识别算法研究

摘要：

针对遥感图像小目标检测精度不高、目标密集不易检测等问题,提出了一种改进YOLOv5s的遥感图像检测算法.在数据增强模块中通过增加垂直旋转方法解决了图像角度不同造成差异的问题;通过在YOLOv5s模型的Backbone模块中引入注意力机制使模型网络更加关注重点信息,以提高目标检测的精度.该算法在TGRS-HRRSD-Dataset数据集上进行实验,召回率达到 99.8％,map@0.5 达到 98.8％,精确率达到75.6％,各项指标有明显提高.

关键词：遥感图像、注意力机制、深度学习、YOLOv5s、垂直旋转

所属期刊栏目：44

分类号：TP391.4(计算技术、计算机技术)

资助基金：国家自然科学基金;中国高校产学研创新基金新一代信息技术创新项目

在线出版日期：2023-05-22（万方平台首次上网日期，不代表论文的发表时间）

页数：共7页

页码：117-123