ResearchBib Share Your Research, Maximize Your Social Impacts
Sign for Notice Everyday Sign up >> Login

Research on Recyclable Garbage Classification Algorithm Based on Attention Mechanism

Journal: International Journal of Science and Research (IJSR) (Vol.9, No. 10)

Publication Date:

Authors : ;

Page : 1330-1333

Keywords : Recyclable garbage classification; Image classification algorithm; Convolutional Neural Network; Attention mechanism;

Source : Downloadexternal Find it from : Google Scholarexternal


With the rapid development of global productivity levels, the problem of garbage disposal is getting more and more serious. Garbage classification is an important step to realize garbage reduction, harmlessness and resource utilization. With the increase in types and quantities of garbage, traditional garbage classification image algorithms can no longer meet the accuracy requirements of garbage identification. This paper proposes a ResNet18 convolutional neural network model based on the attention mechanism for the classification of recyclable garbage. The attention module is added after convolution, so that the model can pay more attention to the important information in the feature map. The model can automatically extract the characteristics of garbage for classification, including: glass, metal, plastic and paper. Experimental results show that the algorithm has an accuracy rate of 92 % in the classification of recyclable waste, which can effectively classify recyclable waste.

Last modified: 2021-06-28 17:13:38