基于改进YOLO的多尺度聚合遥感图像小目标检测算法

邝先验 ,  王星星 ,  王龙锋 ,  张祖梁

科学技术与工程 ›› 2025, Vol. 25 ›› Issue (20) :8560-8570

PDF (21886KB)
科学技术与工程 ›› 2025, Vol. 25 ›› Issue (20) :8560-8570 DOI: 10.12404/j.issn.1671-1815.2405587
自动化技术、计算机技术

基于改进YOLO的多尺度聚合遥感图像小目标检测算法

作者信息 +

Small Target Detection Algorithm for Multi-scale Aggregate Remote Sensing Images Based on Improved YOLO

Author information +
文章历史 +

摘要

针对目前遥感图像小目标检测任务中易出现漏检和误检的问题,提出一种SCS-YOLO[SMCA + CSC + SIoU(shape-aware intersection over union loss)-you only look once]的遥感图像小目标检测算法。首先,针对遥感图像中目标小而聚集的问题,构建空间多尺度卷积注意力(spatial multi-scale convolutional attention, SMCA),提升模型对空间和通道信息的特征提取能力;其次,针对深层网络传递时小目标语义信息容易丢失的问题,设计聚合亚像素卷积(concentrated sub-pixel convolution, CSC),采用多尺度聚合特征提取方法,增强了网络对语义信息的提取能力;最后,将SIoU损失函数替代原模型中的CIoU(complete intersection over union loss)损失函数,加快了网络的收敛速度。SCS-YOLO模型在RSOD和NWPU VHR-10数据集上,平均精确率的平均值(mAP)分别达到97%和90.9%,相较于原模型分别提升了2.2%和2.7%,可见该方法在遥感图像小目标检测任务中的有效性。

Abstract

In order to solve the problems of missed detection and false detection in the current remote sensing image small target detection task, a SMCA + CSC + shape-aware intersection over union loss (SIoU)-you only look once (SCS-YOLO) remote sensing image small target detection algorithm was proposed. Firstly, in response to the problem of small and clustered targets in remote sensing images, a spatial multi-scale convolutional attention module(SMCA) was constructed to improve the model's feature extraction ability of spatial and channel information. Secondly, in order to solve the problem that the semantic information of small targets was easy to be lost during deep network transmission, the aggregation subpixel convolution module concentrated sub-pixel convolution (CSC) was designed, and the multi-scale aggregation feature extraction method was used to enhance the ability of the network to extract semantic information. Finally, the SIoU loss function was used to replace the complete intersection over union loss(CIoU) loss function in the original model, which accelerated the convergence speed of the network. The average of the average precision(mAP) of the SCS-YOLO model reaches 97% and 90.9% on the RSOD and NWPU VHR-10 datasets, respectively, which is 2.2% and 2.7% higher than that of the original model, which shows the effectiveness of the method in the small target detection task of remote sensing images.

关键词

遥感图像 / SCS-YOLO / 小目标 / 注意力 / 聚合亚像素卷积 / SIoU

Key words

remote sensing images / SMCA + CSC + SIoU you only look once(SCS-YOLO) / small target / attention / aggregated sub-pixel convolution / SIoU

引用本文

引用格式 ▾
邝先验, 王星星, 王龙锋, 张祖梁. 基于改进YOLO的多尺度聚合遥感图像小目标检测算法[J]. 科学技术与工程, 2025, 25(20): 8560-8570 DOI:10.12404/j.issn.1671-1815.2405587
KUANG Xian-yan, WANG Xing-xing, WANG Long-feng, ZHANG Zu-liang. Small Target Detection Algorithm for Multi-scale Aggregate Remote Sensing Images Based on Improved YOLO[J]. Science Technology and Engineering, 2025, 25(20): 8560-8570 DOI:10.12404/j.issn.1671-1815.2405587

登录浏览全文

4963

注册一个新账户 忘记密码

参考文献

基金资助

国家自然科学基金(51268017)

PDF (21886KB)

46

访问

0

被引

详细

导航
相关文章

/

〈 〉