首页|基于YOLOv7-RS的遥感图像目标检测研究

基于YOLOv7-RS的遥感图像目标检测研究

扫码查看
针对遥感图像目标检测过程中存在的背景复杂、目标特征不明显、小目标排列密集的问题,基于YOLOv7 算法,提出了一种改进的遥感图像目标检测算法YOLOv7-RS(YOLOv7-Remote Sensing),提高了遥感图像的目标检测精度.首先,向特征提取网络中融合SimAM减少背景噪声的干扰;其次,提出了D-ELAN网络增强遥感目标的特征提取能力;再次,利用SIOU损失函数以提高算法模型的收敛速度;最后,优化了正负样本分配策略,改善了遥感图像中小目标密集排列时的漏检问题.实验结果表明,YOLOv7-RS在NWPU VHR-10 和DOTA数据集上的mAP达到 95.4%和74.1%,相较于其他主流算法有了明显提升.
Research on object detection in remote sensing image based on YOLOv7-RS
Aiming at the problems of complex background,obscure object features and dense array of small targets in remote sensing image target detection,we propose an improved remote sensing image target detection algorithm Yolov7-RS(Yolov7-Re-mote Sensing)based on the YOLOv7 algorithm,which improves the target detection accuracy of remote sensing image.Firstly,SimAM is integrated into feature extraction network to reduce the interference of background noise.Secondly,D-ELAN network enhanced feature extraction capability of remote sensing objects is proposed.Thirdly,SIOU loss function is used to improve the convergence rate of the algorithm model.Finally,the allocation strategy of positive and negative samples is optimized to improve the problem of missing detection when small objects are densely arranged in remote sensing images.Experimental results show that the mAP of YOLOv7-RS on NWPU VHR-10 data sets and DOTA data sets reaches 95.4%and 74.1%,which is significantly im-proved compared with other mainstream algorithms.

remote sensing imagetarget detectionYOLOv7-RSSimAMD-ELANSIOU

梁琦、杨晓文

展开 >

武警山西总队参谋部,山西 太原 030012

中北大学 计算机科学与技术学院, 山西 太原 030051

中北大学 计算机科学与技术学院,山西 太原 030051

机器视觉与虚拟现实山西省重点实验室, 山西 太原 030051

山西省视觉信息处理及智能机器人工程研究中心, 山西 太原 030051

展开 >

遥感图像 目标检测 YOLOv7-RS SimAM D-ELAN SIOU

山西省自然科学基金

202203021222027

2024

网络安全与数据治理
华北计算机系统工程研究所(中国电子信息产业集团有限公司第六研究所)

网络安全与数据治理

影响因子:0.348
ISSN:2097-1788
年,卷(期):2024.43(1)
  • 20