Mathematical Biosciences and Engineering (Jan 2023)

Research on deep learning garbage classification system based on fusion of image classification and object detection classification

  • Zhongxue Yang,
  • Yiqin Bao,
  • Yuan Liu ,
  • Qiang Zhao,
  • Hao Zheng,
  • YuLu Bao

DOI
https://doi.org/10.3934/mbe.2023219
Journal volume & issue
Vol. 20, no. 3
pp. 4741 – 4759

Abstract

Read online

With the development of national economy, the output of waste is also increasing. People's living standards are constantly improving, and the problem of garbage pollution is increasingly serious, which has a great impact on the environment. Garbage classification and processing has become the focus of today. This topic studies the garbage classification system based on deep learning convolutional neural network, which integrates the garbage classification and recognition methods of image classification and object detection. First, the data sets and data labels used are made, and then the garbage classification data are trained and tested through ResNet and MobileNetV2 algorithms, Three algorithms of YOLOv5 family are used to train and test garbage object data. Finally, five research results of garbage classification are merged. Through consensus voting algorithm, the recognition rate of image classification is improved to 2%. Practice has proved that the recognition rate of garbage image classification has been increased to about 98%, and it has been transplanted to the raspberry pie microcomputer to achieve ideal results.

Keywords