计算机技术与发展2024，Vol.34Issue(1)：77-82,6.DOI:10.3969/j.issn.1673-629X.2024.01.012
基于航空图像的目标检测算法Trans_YOLOv5
Target Detection Algorithm Trans_YOLOv5 Based on Aerial Image
摘要
Abstract
Compared with the detection algorithm of natural images,there are problems such as random target angle,sharp change of target scale,dense small targets,and complex image background in aerial image target detection.Trans-YOLOv5 algorithm suitable for aerial image detection is proposed to solve this series of problems.Modifying the data preprocessing module and post-processing method in the YOLOv5 algorithm to add the processing of a target angle label to make it suitable for aerial images with random target angles.CSL(Circular Smooth Label)is introduced to transform the label angle regression issue into a classification issue about the problem of boundary problems.Regarding the issue of small target detection in aerial images,we integrate Swin Transformer into the YOLOv5 framework to capture global semantic information,which improve the detection effect of the model on small targets,and cooperate with the attention mechanism module to improve the global representation ability,so that the network model pays more attention to the target object to be detected.The experimental results on the DOTAv2.0 dataset validate the effectiveness of the proposed method.The detection results reach 60.98%mAP,which is 10.85 percentage points higher than that of the original YOLOv5 algorithm and 2.01 per-centage points higher than the competition results published on the official website.关键词
小目标检测/航空图像/YOLOv5/圆形平滑标签/Swin TransformerKey words
small target detection/aerial images/YOLOv5/circular smooth label/Swin Transformer分类
信息技术与安全科学引用本文复制引用
文青,伍欣,敖斌,李宽,殷建平..基于航空图像的目标检测算法Trans_YOLOv5[J].计算机技术与发展,2024,34(1):77-82,6.基金项目
国家重点研发计划(2018YFB1003203) （2018YFB1003203）
国家自然科学基金项目(62206054) （62206054）
