LC-YOLO: A Lightweight Model with Efficient Utilization of Limited Detail Features for Small Object Detection
-
Published:2023-03-01
Issue:5
Volume:13
Page:3174
-
ISSN:2076-3417
-
Container-title:Applied Sciences
-
language:en
-
Short-container-title:Applied Sciences
Author:
Cui Menghua12ORCID, Gong Guoliang12, Chen Gang12, Wang Hongchang12ORCID, Jin Min12, Mao Wenyu12ORCID, Lu Huaxiang12345
Affiliation:
1. Institute of Semiconductors, Chinese Academy of Sciences, Beijing 100083, China 2. University of Chinese Academy of Sciences, Beijing 100049, China 3. Materials and Optoelectronics Research Center, University of Chinese Academy of Sciences, Beijing 100049, China 4. College of Microelectronics, University of Chinese Academy of Sciences, Beijing 100049, China 5. Semiconductor Neural Network Intelligent Perception and Computing Technology Beijing Key Laboratory, Beijing 100083, China
Abstract
The limited computing resources on edge devices such as Unmanned Aerial Vehicles (UAVs) mean that lightweight object detection algorithms based on convolution neural networks require significant development. However, lightweight models are challenged by small targets with few available features. In this paper, we propose an LC-YOLO model that uses detailed information about small targets in each layer to improve detection performance. The model is improved from the one-stage detector, and contains two optimization modules: Laplace Bottleneck (LB) and Cross-Layer Attention Upsampling (CLAU). The LB module is proposed to enhance shallow features by integrating prior information into the convolutional neural network and maximizing knowledge sharing within the network. CLAU is designed for the pixel-level fusion of deep features and shallow features. Under the combined action of these two modules, the LC-YOLO model achieves better detection performance on the small object detection task. The LC-YOLO model with a parameter quantity of 7.30M achieves an mAP of 94.96% on the remote sensing dataset UCAS-AOD, surpassing the YOLOv5l model with a parameter quantity of 46.61M. The tiny version of LC-YOLO with 1.83M parameters achieves 94.17% mAP, which is close to YOLOv5l. Therefore, the LC-YOLO model can replace many heavyweight networks to complete the small target high-precision detection task under limited computing resources, as in the case of mobile edge-end chips such as UAV onboard chips.
Funder
National Natural Science Foundation of China Strategic Priority Research Program of the Chinese Academy of Sciences
Subject
Fluid Flow and Transfer Processes,Computer Science Applications,Process Chemistry and Technology,General Engineering,Instrumentation,General Materials Science
Reference34 articles.
1. Simonyan, K., and Zisserman, A. (2014). Very deep convolutional networks for large-scale image recognition. Computer Science. arXiv. 2. Szegedy, C., Liu, W., Jia, Y., Sermanet, P., and Rabinovich, A. (2014, January 7–12). Going deeper with convolutions. Proceedings of the IEEE Computer Society, Boston, MA, USA. 3. He, K., Zhang, X., Ren, S., and Sun, J. (July, January 26). Deep residual learning for image recognition. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA. 4. Redmon, J., and Farhadi, A. (2018). Yolov3: An incremental improvement. arXiv. 5. Wang, C.Y., Liao, H., Wu, Y.H., Chen, P.Y., and Yeh, I.H. (2020, January 14–19). CSPNet: A New Backbone that can Enhance Learning Capability of CNN. Proceedings of the 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW), Seattle, WA, USA.
Cited by
3 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献
|
|