UN-YOLOv5s: A UAV-Based Aerial Photography Detection Algorithm
Author:
Guo Junmei1, Liu Xingchen1, Bi Lingyun1, Liu Haiying1, Lou Haitong1
Affiliation:
1. The School of Information and Automation Engineering, Qilu University of Technology (Shandong Academy of Sciences), Jinan 250353, China
Abstract
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%.
Subject
Electrical and Electronic Engineering,Biochemistry,Instrumentation,Atomic and Molecular Physics, and Optics,Analytical Chemistry
Reference37 articles.
1. Fahlstrom, P.G., Gleason, T.J., and Sadraey, M.H. (2022). Introduction to UAV Systems, John Wiley & Sons. 2. Alam, S.S., Chakma, A., Rahman, M.H., Bin Mofidul, R., Alam, M.M., Utama, I.B.K.Y., and Jang, Y.M. (2023). RF-Enabled Deep-Learning-Assisted Drone Detection and Identification: An End-to-End Approach. Sensors, 23. 3. Tsoukalas, A., Xing, D., Evangeliou, N., Giakoumidis, N., and Tzes, A. (2021, January 15–18). Deep learning assisted visual tracking of evader-UAV. Proceedings of the 2021 International Conference on Unmanned Aircraft Systems (ICUAS), Athens, Greece. 4. Swarm Reconnaissance Drone System for Real-Time Object Detection Over a Large Area;Moon;IEEE Access,2023 5. Lou, H.T., Duan, X.H., Guo, J.M., Liu, H.Y., Gu, J.S., Bi, L.Y., and Chen, H.N. (2023). DC-YOLOv8: Small Size Object Detection Algorithm Based on Camera Sensor. Electronics, 12.
Cited by
4 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献
|
|