UN-YOLOv5s:一种基于无人机的航空摄影检测算法。
UN-YOLOv5s: A UAV-Based Aerial Photography Detection Algorithm.
作者信息
Guo Junmei, Liu Xingchen, Bi Lingyun, Liu Haiying, Lou Haitong
机构信息
The School of Information and Automation Engineering, Qilu University of Technology (Shandong Academy of Sciences), Jinan 250353, China.
出版信息
Sensors (Basel). 2023 Jun 26;23(13):5907. doi: 10.3390/s23135907.
With the progress of science and technology, artificial intelligence is widely used in various disciplines and has produced amazing results. The research of the target detection algorithm has significantly improved the performance and role of unmanned aerial vehicles (UAVs), and plays an irreplaceable role in preventing forest fires, evacuating crowded people, surveying and rescuing explorers. At this stage, the target detection algorithm deployed in UAVs has been applied to production and life, but making the detection accuracy higher and better adaptability is still the motivation for researchers to continue to study. In aerial images, due to the high shooting height, small size, low resolution and few features, it is difficult to be detected by conventional target detection algorithms. In this paper, the UN-YOLOv5s algorithm can solve the difficult problem of small target detection excellently. The more accurate small target detection (MASD) mechanism is used to greatly improve the detection accuracy of small and medium targets, The multi-scale feature fusion (MCF) path is combined to fuse the semantic information and location information of the image to improve the expression ability of the novel model. The new convolution SimAM residual (CSR) module is introduced to make the network more stable and focused. On the VisDrone dataset, the mean average precision (mAP) of UAV necessity you only look once v5s(UN-YOLOv5s) is 8.4% higher than that of the original algorithm. Compared with the same version, YOLOv5l, the mAP is increased by 2.2%, and the Giga Floating-point Operations Per Second (GFLOPs) is reduced by 65.3%. Compared with the same series of YOLOv3, the mAP is increased by 1.8%, and GFLOPs is reduced by 75.8%. Compared with the same series of YOLOv8s, the detection accuracy of the mAP is improved by 1.1%.
随着科学技术的进步,人工智能在各个学科中得到广泛应用,并取得了惊人的成果。目标检测算法的研究显著提高了无人机(UAV)的性能和作用,在预防森林火灾、疏散拥挤人群、勘测和救援探险者方面发挥着不可替代的作用。现阶段,部署在无人机上的目标检测算法已应用于生产生活,但提高检测精度和更好的适应性仍是研究人员继续研究的动力。在航空图像中,由于拍摄高度高、尺寸小、分辨率低且特征少,传统目标检测算法难以进行检测。本文中,UN-YOLOv5s算法能够出色地解决小目标检测难题。采用更精确的小目标检测(MASD)机制大幅提高了中小目标的检测精度,结合多尺度特征融合(MCF)路径融合图像的语义信息和位置信息,提高了新型模型的表达能力。引入新的卷积SimAM残差(CSR)模块,使网络更加稳定和聚焦。在VisDrone数据集上,无人机你只看一次v5s(UN-YOLOv5s)的平均精度均值(mAP)比原算法高8.4%。与同一版本的YOLOv5l相比,mAP提高了2.2%,每秒千兆浮点运算次数(GFLOPs)降低了65.3%。与同系列的YOLOv3相比,mAP提高了1.8%,GFLOPs降低了75.8%。与同系列的YOLOv8s相比,mAP的检测精度提高了1.1%。