沈阳航空航天大学学报2024，Vol.41Issue(2)：37-46,10.DOI:10.3969/j.issn.2095-1248.2024.02.005
基于YOLOv5的无人机航拍小目标检测模型
A small target detection model for UAV aerial photography based on YOLOv5
摘要
Abstract
In order to solve the problems of high missed detection rate and low detection success rate in UAV small target detection,a small target detection algorithm based on YOLOv5 was proposed.Firstly,the swin transformer module was integrated into the backbone structure and the neck structure respectively,which improved the accuracy of target detection on the basis of reducing the computation-al cost,and could adapt to the detection of small target in UAV aerial photography.Secondly,the con-volutional attention module(CBAM)was introduced to enhance the network's attention for small tar-get features.Finally,the original loss function CIoU was replaced by the SIOU loss function,and the weights of high-quality samples were emphasized to accelerate convergence and improve the regression accuracy.Experimental results show that the detection accuracy on Visdrone2019 dataset is 35.3%after model optimization,which is 5.2%higher than that of YOLOv5.Compared with other classical and ad-vanced algorithms,SWCBSI-YOLO algorithm performs well and meets the detection requirements of small targets for UAV aerial photography.关键词
无人机航拍图像/小目标检测/YOLOv5/transformer/注意力机制/损失函数Key words
UAV aerial photography/small target detection/YOLOv5/transformer/attention mecha-nism/loss function分类
信息技术与安全科学引用本文复制引用
石祥滨,赵芮同..基于YOLOv5的无人机航拍小目标检测模型[J].沈阳航空航天大学学报,2024,41(2):37-46,10.基金项目
国家自然科学基金(项目编号:61170185) （项目编号:61170185）
