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.
This is a preview of subscription content, log in via an institution.
Buying options
Tax calculation will be finalised at checkout
Purchases are for personal use only
Learn about institutional subscriptionsReferences
Mittal, G., Yagnik, K.B., Garg, M., Krishnan, N.C.: Spotgarbage: smartphone app to detect garbage using deep learning. In: Proceedings of the 2016 ACM International Joint Conference on Pervasive and Ubiquitous Computing, pp. 940–945 (2016)
Yang, M.: Classification of trash for recyclability status. CS229 project report, March 2016
Mao, W.L., Chen, W.C., Wang, C.T., Lin, Y.H.: Recycling waste classification using optimized convolutional neural network. Resour. Conserv. Recycl. 164, 105132 (2021)
Zhang, R., Yin, D., Ding, J., Luo, Y., Liu, W., Yuan, M.: A detection method for low-pixel ratio object. Multimed. Tools Appl. 78(9), 11655–11674 (2019)
Yu, X., Chen, Z., Zhang, S., Zhang, T.: A street rubbish detection algorithm based on Sift and RCNN. In: MIPPR 2017: Automatic Target Recognition and Navigation, vol. 10608, p. 106080F. International Society for Optics and Photonics, February 2018
Howard, A.G., et al.: Mobilenets: Efficient convolutional neural networks for mobile vision applications. arXiv preprint arXiv:1704.04861 (2017)
Sandler, M., Howard, A., Zhu, M., Zhmoginov, A., Chen, L.C.: Mobilenetv2: inverted residuals and linear bottlenecks. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 4510–4520 (2018)
Howard, A., et al.: Searching for mobilenetv3. In: Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 1314–1324 (2019)
Han, K., Wang, Y., Tian, Q., Guo, J., Xu, C., Xu, C.: Ghostnet: More features from cheap operations. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 1580–1589 (2020)
Zhang, X., Zhou, X., Lin, M., Sun, J.: Shufflenet: an extremely efficient convolutional neural network for mobile devices. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 6848–6856 (2018)
Ma, N., Zhang, X., Zheng, H.T., Sun, J.: Shufflenet v2: practical guidelines for efficient cnn architecture design. In Proceedings of the European Conference on Computer Vision, pp. 116–131 (2018)
Redmon, J., Divvala, S., Girshick, R., Farhadi, A.: You only look once: Unified, real-time object detection. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 779–788 (2016)
Redmon, J., Farhadi, A.: YOLO9000: better, faster, stronger. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 7263–7271 (2017)
Redmon, J., Farhadi, A.: Yolov3: an incremental improvement. arXiv preprint arXiv:1804.02767 (2018)
Bochkovskiy, A., Wang, C.Y., Liao, H.Y.M.: Yolov4: optimal speed and accuracy of object detection. arXiv preprint arXiv:2004.10934 (2020)
Yao, X., Huang, T., Wu, C., Zhang, R., Sun, L.: Towards faster and better federated learning: a feature fusion approach. In: 2019 IEEE International Conference on Image Processing, pp. 175–179, September 2019
Acknowledgments
This work was financially supported by National Natural Science Foundation of China (62176085, 61672204, 62172458), Nature Science Research Project of Anhui province (1908085MF185), Major Scientific Research Projects of Universities of Anhui Province (KJ2019ZD61), Key Projects of Excellent Young Talents Support Program (gxyq2019113), China's Post-doctoral Science Fund (2020M681989), Talent Fund of Hefei University (20RC25).
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2022 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Wang, XF., Wang, JT., Xu, LX., Tan, M., Yang, J., Tang, Yy. (2022). Garbage Classification Detection Model Based on YOLOv4 with Lightweight Neural Network Feature Fusion. In: Huang, DS., Jo, KH., Jing, J., Premaratne, P., Bevilacqua, V., Hussain, A. (eds) Intelligent Computing Methodologies. ICIC 2022. Lecture Notes in Computer Science(), vol 13395. Springer, Cham. https://doi.org/10.1007/978-3-031-13832-4_36
Download citation
DOI: https://doi.org/10.1007/978-3-031-13832-4_36
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-13831-7
Online ISBN: 978-3-031-13832-4
eBook Packages: Computer ScienceComputer Science (R0)