UM  > Faculty of Science and Technology
Residential Collegefalse
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 Name18th International Conference on Intelligent Computing, ICIC 2022
Source PublicationLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume13395 LNAI
Pages435-447
Conference Date07-11 August 2022
Conference PlaceXi'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.

KeywordYolov4-lff Lightweight Neural Networks Garbage Classification Detection Feature Fusion Deep Learning
DOI10.1007/978-3-031-13832-4_36
URLView the original
Indexed ByCPCI-S
Language英語English
WOS Research AreaComputer Science
WOS SubjectComputer Science, Artificial Intelligence ; Computer Science, Interdisciplinary Applications
WOS IDWOS:000870337200036
Scopus ID2-s2.0-85137262265
Fulltext Access
Citation statistics
Document TypeConference paper
CollectionFaculty of Science and Technology
DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE
Corresponding AuthorYuan-yan Tang
Affiliation1.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 AffilicationFaculty 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.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Xiao-Feng Wang]'s Articles
[Jian-Tao Wang]'s Articles
[Li-Xiang Xu]'s Articles
Baidu academic
Similar articles in Baidu academic
[Xiao-Feng Wang]'s Articles
[Jian-Tao Wang]'s Articles
[Li-Xiang Xu]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Xiao-Feng Wang]'s Articles
[Jian-Tao Wang]'s Articles
[Li-Xiang Xu]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.