改进YOLOv6的遥感图像目标检测算法

作     者：许德刚 王再庆 邢奎杰 郭奕欣 XU Degang;WANG Zaiqing;XING Kuijie;GUO Yixin

作者机构：河南工业大学信息科学与工程学院郑州450001 河南工业大学粮食信息处理与控制教育部重点实验室郑州450001 湖南工程学院计算机与通信学院湖南湘潭411104 

出 版 物：《计算机工程与应用》 (Computer Engineering and Applications)

年 卷 期：2024年第60卷第3期

页      面：119-128页

核心收录：

学科分类：081203[工学-计算机应用技术] 08[工学] 0835[工学-软件工程] 0812[工学-计算机科学与技术（可授工学、理学学位）] 

基　　金：国家重点研发计划(2017YFD0401003-4) 湖南省自然科学基金(2021JJ50114) 

主　　题：遥感图像 目标检测 注意力机制 多尺度目标 YOLOv6 

摘      要：针对遥感图像背景复杂、目标普遍比较小且呈多尺度分布所导致的目标检测精度较低的问题,提出了一种改进YOLOv6的遥感图像目标检测算法。在骨干网络引入一种坐标注意力模块,以提高复杂背景下模型的特征提取能力和目标定位能力。提出一种上下文增强模块,使模型获取丰富的上下文信息,从而提升模型提取多尺度目标细节信息的能力。为了实现不同尺度特征的自适应融合,通过在颈网络引入一种自适应空间特征融合,提升了多尺度目标尤其是小目标的检测精度。将所提改进算法在遥感图像公开数据集DOTA-v1.0上进行训练并测试,实验结果表明,改进算法的收敛速度与收敛精度均优于原算法,其中AP值达到了54.6%,相比原算法提高了1.4个百分点,同时相比于一些其他目前先进的目标检测算法在精度和速度上均有提升,证明了改进算法的有效性。