Lightweight Small Target Detection Network Based on UAV

CONG Yuhua; HE Xiao; XING Changda; CHENG Xuming; TANG Xin; WANG Zhisheng; OUYANG Quan

doi:10.15892/j.cnki.djzdxb.2022.06.002

Journal of Projectiles, Rockets, Missiles and Guidance >

2022 , Vol. 42 >Issue 6: 6 - 12

DOI: https://doi.org/10.15892/j.cnki.djzdxb.2022.06.002

Lightweight Small Target Detection Network Based on UAV

CONG Yuhua ¹^,² ,
HE Xiao ² ,
XING Changda ¹^,³ ,
CHENG Xuming ¹ ,
TANG Xin ¹ ,
WANG Zhisheng ¹ ,
OUYANG Quan ¹

Expand

¹ College of Automation Engineering, Nanjing University of Aeronautics and Astronautics, Nanjing 210006, China
² Zijin College Nanjing University of Science and Technology, Nanjing 210023, China
³ Shenzhen Research Institute of Nanjing University of Aeronautics and Astronautics, Shenzhen 518063, Guangdong, China

Received date: 2022-06-23

Online published: 2025-02-25

Fold

Abstract

When target detection is carried out on the UAV terminal, there are some problems such as limited computing power of the airborne embedded equipment, small target, complex background and low image resolution. Therefore, the target detection network adopts the YOLO system framework, and on this basis, the target detection network is improved in lightweight and accuracy.The SPP module is introduced into the backbone feature extraction network part, the attention mechanism is introduced into the feature enhancement network part, and the decoupling head mechanism is used in the feature output part. The comparison test shows that the improved algorithm has real-time and high accuracy, and is suitable for the application of unmanned aerial vehicles.

Key words： UAV; YOLO; lightweight; attentional mechanism; small target detection

Cite this article

CONG Yuhua , HE Xiao , XING Changda , CHENG Xuming , TANG Xin , WANG Zhisheng , OUYANG Quan . Lightweight Small Target Detection Network Based on UAV[J]. Journal of Projectiles, Rockets, Missiles and Guidance, 2022 , 42(6) : 6 -12 . DOI: 10.15892/j.cnki.djzdxb.2022.06.002

[an error occurred while processing this directive]

References

Publishing order | Descend order by publishing year | Descend order by cited within

[1]	REDMON J, FARHADI A. YOLOv3: An incremental improvement[D]. Washington: Washington University, 2018.

[2]	陈灏然. 基于卷积神经网络的小目标检测算法研究[D]. 无锡: 江南大学, 2021.

[3]	石鑫鑫. 改进的卷积神经网络算法在农作物病害检测中的应用[D]. 长春: 长春工业大学, 2021.

[4]	BOCHKOVSKIY A, WANG C Y, LIAO H Y M. YOLOv4: Optimal speed and accuracy of object detection[EB/OL].(2004-10-23)[2022-08-17]. http://https://arxiv.org/abs/2004.10934.

[5]	WANG C Y, LIAO H Y M, WU Y H, et al. CSPNet: a new backbone that can enhance learning capability of CNN[C]// IEEE. Proceedings of the IEEE/CVF conference on computer vision and pattern recognition workshops. New York: IEEE, 2020: 390-391.

[6]	HE K, ZHANG X, REN S, et al. Deep residual learning for image recognition[C]// IEEE. Proceedings of the IEEE conference on computer vision and pattern recognition. New York: IEEE, 2016: 770-778.

[7]	HE K, ZHANG X, REN S, et al. Spatial pyramid pooling in deep convolutional networks for visual recognition[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2015, 37(9): 1904-1916. DOI PMID

[8]	TAN M, PANG R, LE Q V. Efficientdet: Scalable and efficient object detection[C]// IEEE. Proceedings of Computer Vision (ECCV). New York: IEEE, 2018: 3-19.

[9]	LIU S, QI L, QIN H, et al. Path aggregation network or instance segmentation[C]// IEEE.Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. New York: IEEE, 2018: 8759-8768.

[10]	WOO S, PARK J, LEE J Y, et al. Cbam: Convolutional block attention module[C]// IEEE. Proceedingsof the European conference on learning for image recognition. New York: IEEE, 2016:770-778.

[11]	WANG H, WANG J, BAI K, et al. Centered multi-task generative adversarial network for small object detection[J]. Sensors, 2021, 21(15): 5194.

[12]	ZHU P, WEN L, DU D, et al. Detection and tracking meet drones challenge[J]. IEEE Transactions on Pattern Analysis & Machine Intelligence, 2021 (1): 11-17.

Options

Outlines