[an error occurred while processing this directive] [an error occurred while processing this directive]
[an error occurred while processing this directive]
石祥滨(1963-),男,辽宁沈阳人,教授,博士,主要研究方向:计算机视觉,E-mail:sxb@sau.edu.cn。 |
收稿日期: 2023-12-20
网络出版日期: 2024-05-29
基金资助
国家自然科学基金(61170185)
A small target detection model for UAV aerial photography based on YOLOv5
Received date: 2023-12-20
Online published: 2024-05-29
针对无人机小目标检测中漏检率高、检测成功率低等问题,提出一种基于YOLOv5的小目标检测算法。首先,分别在backbone结构和neck结构中,融合swin transformer模块,在减少计算成本的基础上,提高目标检测的准确率,以适应无人机航拍小目标检测;其次,引入卷积注意力模块(convolutional block attention module,CBAM),以增强网络对小目标特征的关注度;最后,将原始损失函数CIoU替换为SIoU损失函数,强调高质量样本权重加速收敛,提高回归精度。实验结果表明,经过模型优化,在Visdrone2019数据集上的检测精度为35.3%,与YOLOv5相比,提升了5.2%;相较于其他经典及先进算法,SWCBSI-YOLO算法表现良好,满足针对无人机航拍小目标的检测要求。
石祥滨 , 赵芮同 . 基于YOLOv5的无人机航拍小目标检测模型[J]. 沈阳航空航天大学学报, 2024 , 41(2) : 37 -46 . DOI: 10.3969/j.issn.2095-1248.2024.02.005
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 computational cost, and could adapt to the detection of small target in UAV aerial photography.Secondly, the convolutional block attention module (CBAM) was introduced to enhance the network’s attention for small target 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 advanced algorithms,SWCBSI-YOLO algorithm performs well and meets the detection requirements of small targets for UAV aerial photography.
1 |
陈旭.基于深度学习的无人机图像目标检测算法研究[D].杭州:杭州电子科技大学,2022.
|
2 |
江波,屈若锟,李彦冬,等.基于深度学习的无人机航拍目标检测研究综述 [J].航空学报,2021,42 (4):137-151.
|
3 |
|
4 |
|
5 |
|
6 |
王龙博,刘建辉,张贝贝,等.利用注意力机制融合的YOLOv5遥感图像目标检测[J].信息工程大学学报,2023,24(4):438-446.
|
7 |
赵倩,王成龙,郭彤.一种模型轻量化设计的遥感小目标检测方法:CN112329721A[P].2021-02-05.
|
8 |
|
9 |
|
10 |
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 |
|
17 |
|
18 |
|
19 |
|
20 |
|
21 |
|
22 |
|
/
〈 |
|
〉 |