[1]牛为华,郭迅.融合分块注意力与小波特征聚合的遥感图像目标检测算法[J].智能系统学报,2026,21(3):763-775.[doi:10.11992/tis.202507006]
NIU Weihua,GUO Xun.Remote sensing object detection algorithm integrating block attention and wavelet feature aggregation[J].CAAI Transactions on Intelligent Systems,2026,21(3):763-775.[doi:10.11992/tis.202507006]
点击复制
《智能系统学报》[ISSN 1673-4785/CN 23-1538/TP] 卷:
21
期数:
2026年第3期
页码:
763-775
栏目:
学术论文—人工智能基础
出版日期:
2026-05-05
- Title:
-
Remote sensing object detection algorithm integrating block attention and wavelet feature aggregation
- 作者:
-
牛为华1,2, 郭迅1
-
1. 华北电力大学 计算机系, 河北 保定 071003;
2. 复杂能源系统与智能计算教育部工程研究中心, 河北 保定 071003
- Author(s):
-
NIU Weihua1,2, GUO Xun1
-
1. Department of Computer Science, North China Electric Power University, Baoding 071003, China;
2. Engineering Research Center of Intelligent Computing for Complex Energy System, Ministry of Education, Baoding 071003, China
-
- 关键词:
-
目标检测; 遥感影像; 小波变换; YOLO; 注意力机制; 小目标检测; 特征提取; 特征融合
- Keywords:
-
object detection; remote sensing imagery; wavelet transform; YOLO; attention mechanism; small object detection; feature extraction; feature fusion
- 分类号:
-
TP391.41
- DOI:
-
10.11992/tis.202507006
- 文献标志码:
-
2026-3-9
- 摘要:
-
针对遥感图像中目标背景复杂和小目标物体检测困难等问题,提出融合分块注意力与小波特征聚合的遥感图像目标检测算法。在主干网络中引入层级分块注意力模块,采用分块下采样操作进行下采样操作,同时融合坐标注意力机制,缓解了细粒度信息丢失的问题;设计融合小波增强的对比驱动特征聚合模块,采用Haar小波变换分离图像高低频信息,并构建局部注意力机制,增强模型对目标边缘与纹理细节的感知能力,抑制复杂背景的干扰;构建浅层注意力增强检测头,融合LSKA(large selective kernel attention)注意力机制增强模型对小目标物体的检测精度。实验结果表明,改进后算法在ShipRSImageNet与VisDrone2019数据集上的mAP50分别达到87.3%与35.5%,相比原始模型分别提升5.9%和4.1%。改进后的模型在遥感图像目标检测任务上的性能得到了有效的提升。
- Abstract:
-
A remote sensing image object detection algorithm is proposed to address the challenges of complex backgrounds and small object detection. A hierarchical split attention block attention module is introduced into the backbone network, utilizing block-based downsampling and coordinate attention to mitigate fine-grained information loss. Using the Haar wavelet transform, a wavelet-enhanced contrast-driven feature aggregation module separates high- and low-frequency information, while a local attention mechanism enhances edge and texture perception, suppressing background interference. A shallow enhancement attention detection head with the LSKA attention mechanism improves small object detection accuracy. Experimental results show that the algorithm achieves an mAP50 of 87.3% on the ShipRSImageNet dataset and 35.5% on the VisDrone2019 dataset, with improvements of 5.9% and 4.1%, respectively, over the original model. The improved model considerably enhances performance in remote sensing image object detection.
更新日期/Last Update:
1900-01-01