Residential College | false |
Status | 已發表Published |
Garbage Classification Detection Model Based on YOLOv4 with Lightweight Neural Network Feature Fusion | |
Xiao-Feng Wang1; Jian-Tao Wang1; Li-Xiang Xu1; Ming Tan1; Jing Yang1; Yuan-yan Tang2 | |
2022-08-16 | |
Conference Name | 18th International Conference on Intelligent Computing, ICIC 2022 |
Source Publication | Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics) |
Volume | 13395 LNAI |
Pages | 435-447 |
Conference Date | 07-11 August 2022 |
Conference Place | Xi'an, China |
Abstract | In recent years, global garbage production has increased dramatically, and the garbage has not been treated effectively. To address such problems as the size of the current garbage classification detection model is too large, processing speed is slow, and it is not suitable for deployment to embedded terminals, this paper proposes a YOLOv4 based on lightweight feature fusion (YOLOv4-LFF). The model uses two lightweight neural networks, MobileNetV3 and GhostNet, to execute feature fusion, which is used instead of the CSPDarknet53. It serves to extract preliminary feature information from the images based on the lightweight model. To further reduce the model's size, we replace the standard convolution in PANet with the depthwise separable convolution in the model, which is used for the enhanced feature information extraction work. The final experimental results show that YOLOv4-LFF achieves 93.2% accuracy on the homemade dataset and reduces the number of model parameters to 26.5% of YOLOv4, which significantly reduces the model parameters and memory consumption. Therefore, the YOLOv4-LFF garbage classification detection model meets the requirements of edge computing devices and has theoretical research significance and practicality. |
Keyword | Yolov4-lff Lightweight Neural Networks Garbage Classification Detection Feature Fusion Deep Learning |
DOI | 10.1007/978-3-031-13832-4_36 |
URL | View the original |
Indexed By | CPCI-S |
Language | 英語English |
WOS Research Area | Computer Science |
WOS Subject | Computer Science, Artificial Intelligence ; Computer Science, Interdisciplinary Applications |
WOS ID | WOS:000870337200036 |
Scopus ID | 2-s2.0-85137262265 |
Fulltext Access | |
Citation statistics | |
Document Type | Conference paper |
Collection | Faculty of Science and Technology DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Corresponding Author | Yuan-yan Tang |
Affiliation | 1.Anhui Provincial Engineering Laboratory of Big Data Technology Application for Urban Infrastructure, School of Artificial Intelligence and Big Data, Hefei University, Hefei , 230601, Anhui, China 2.Zhuhai UM Science and Technology Research Institute, FST University of Macau, Macao |
Corresponding Author Affilication | Faculty of Science and Technology |
Recommended Citation GB/T 7714 | Xiao-Feng Wang,Jian-Tao Wang,Li-Xiang Xu,et al. Garbage Classification Detection Model Based on YOLOv4 with Lightweight Neural Network Feature Fusion[C], 2022, 435-447. |
APA | Xiao-Feng Wang., Jian-Tao Wang., Li-Xiang Xu., Ming Tan., Jing Yang., & Yuan-yan Tang (2022). Garbage Classification Detection Model Based on YOLOv4 with Lightweight Neural Network Feature Fusion. Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics), 13395 LNAI, 435-447. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment