基于改进YOLOv8的SAR图像飞机目标检测算法  

An Aircraft Target Detection Algorithm Based on Improved YOLOv8 in SAR Image

作　　者：陈益方 张上[2,3] 冉秀康 王杰 CHEN Yifang;ZHANG Shang;RAN Xiukang;WANG Jie(College of Electrical Engineering and New Energy,China Three Gorges University,Yichang 443002,China;Hubei Province Engineering Technology Research Center for Construction Quality Testing Equipment,China Three Gorges University,Yichang 443002,China;College of Computer and Information,China Three Gorges University,Yichang 443002,China)

机构地区：[1]三峡大学电气与新能源学院,湖北宜昌443002 [2]三峡大学湖北省建筑质量检测装备工程技术研究中心,湖北宜昌443002 [3]三峡大学计算机与信息学院,湖北宜昌443002

出　　处：《电讯技术》2024年第8期1206-1212,共7页Telecommunication Engineering

基　　金：国家级大学生创新创业训练计划(202011075013,202111075012)。

摘　　要：针对合成孔径雷达(Synthetic Aperture Radar, SAR)图像飞机目标检测算法存在模型复杂度较高、检测效果差、泛化能力弱等问题,提出了一种基于改进YOLOv8的SAR图像飞机目标检测算法。首先,针对SAR图像飞机目标较小的特点,剔除大目标检测层,重构特征提取网络和特征融合网络,降低模型计算量。其次,在主干网络引入可变形卷积(Deformable Convolutional Network, DCN),增强特征提取能力;在颈部网络引入全局注意力机制(Global Attention Mechanism, GAM)提高检测精度。最后,采用WIOU(Wise-IoU)损失函数提高收敛速度和回归精度。在SADD数据集(SAR Aircraft Detection Dataset)上实验结果显示,改进算法较原YOLOv8算法模型体积压缩59.66%,参数量降低61.18%,计算量减少18.29%,最高精度提高至98.1%。与其他算法相比,所提算法在保证较高检测精度的情况下大幅降低了模型体积、参数量和计算量,实现了模型复杂度和检测精度的平衡。To solve the problems of high model complexity,poor detection performance,and poor ability of generalization in synthetic aperture radar(SAR)aircraft target detection,a novel algorithm based on improved YOLOv8 is proposed.First,considering the small size of aircraft targets in synthetic aperture radar(SAR)images,the algorithm removes the large object detection layers and reconstructs the feature extraction network and feature fusion network to reduce model computation.Secondly,it introduces deformable convolutions into the backbone network to enhance feature extraction capability and incorporates the global attention mechanism(GAM)into the neck network to improve detection accuracy.Finally,it adopts the Wise-IoU(WIOU)loss function to enhance convergence speed and regression Experimental accuracy.results on the SAR Aircraft Detection(SADD)dataset demonstrate that the proposed algorithm achieves a 59.66%reduction in model size,a 61.18%decrease in parameter quantity,and an 18.29%reduction in computation,while achieving a maximum accuracy of 98.1%.By comparing the proposed algorithm with other approaches,the results indicate a significant reduction in model parameter size,quantity,and computation while maintaining high detection accuracy,validate the balanced trade-off between model complexity and detection accuracy.

关 键 词：合成孔径雷达 飞机目标检测 网络重构 可变形卷积 GAM注意力机制 

分 类 号：TN957.52[电子电信—信号与信息处理]