Research on deep learning image recognition technology in garbage classification

被引:8
|
作者
Guo, Qiang [1 ]
Shi, Yuliang [1 ]
Wang, Shikai [1 ]
机构
[1] Beijing Univ Technol, Fac Informat Technol, Sch Software Engn, Beijing, Peoples R China
关键词
deep learning; convolution neural network; image processing technology; garbage classification; ATTENTION MECHANISMS;
D O I
10.1109/ACCTCS52002.2021.00027
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Nowadays, with the rapid development of industry, people's daily life is becoming more and more rich and diverse. However, while enjoying the pleasure brought by material life, a large number of garbage are produced. They are various, and people lack the knowledge related to garbage classification, which leads to the difficulty of manual or robot classification This paper studies a garbage classification algorithm model based on deep learning convolutional neural network Efficientnet to help identify garbage classification. In this research, data augmentation and normalization are carried out to solve the problem of small amount of data sets and different sizes of pictures. Efficientnet is used to extract the features of images. In order to solve the problem that BN has no obvious effect on small batches in the network, we replace BN with group normalization (GN). In order to prevent some irrelevant information in the image from affecting the training of the model, we add attention mechanism after the output of Efficientnet to emphasize or select the important information of the target processing object, and suppress some irrelevant details, so that the model can focus on the key features and better identify the image; according to the above process, we use softmax to classify the spam image and divide it into four categories (Recyclables, Kitchen garbage , Hazardous garbage ,Other garbage) The results show that the model can effectively extract the features of the input garbage image, and get accurate judgment, and identify the types of garbage. The experimental results show that the average accuracy of the algorithm model is high, and has good classification performance and robustness. In the practical significance of the research, this reliable model can help people quickly know the type of garbage, or can be applied to robot sorting, to help detect the types of garbage for robot judgment and sorting, so it has very important application scenarios and significance.
引用
收藏
页码:92 / 96
页数:5
相关论文
共 50 条
  • [1] Research on Image Recognition Based on Deep Learning Technology
    Zhai, Hao
    [J]. PROCEEDINGS OF THE 2016 4TH INTERNATIONAL CONFERENCE ON ADVANCED MATERIALS AND INFORMATION TECHNOLOGY PROCESSING (AMITP 2016), 2016, 60 : 266 - 270
  • [2] Research on deep learning garbage classification system based on fusion of image classification and object detection classification
    Yang, Zhongxue
    Bao, Yiqin
    Liu, Yuan
    Zhao, Qiang
    Zheng, Hao
    Bao, YuLu
    [J]. MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2023, 20 (03) : 4741 - 4759
  • [3] Application of Image Recognition Technology in Garbage Classification Education Platform
    Chen, Guo
    Wang, Hao
    Zheng, Junhui
    [J]. CONFERENCE PROCEEDINGS OF 2019 5TH INTERNATIONAL CONFERENCE ON CONTROL, AUTOMATION AND ROBOTICS (ICCAR), 2019, : 290 - 294
  • [4] Research on Image Classification and Recognition Technology Based on Machine Learning
    Wang, Yan
    [J]. Applied Mathematics and Nonlinear Sciences, 2024, 9 (01)
  • [5] Application research of image classification algorithm based on deep learning in household garbage sorting
    Wang, Jianfei
    [J]. HELIYON, 2024, 10 (09)
  • [6] Research on Garbage Classification and Recognition Based on Hyperspectral Imaging Technology
    Zhao Dong-e
    Wu Rui
    Zhao Bao-guo
    Chen Yuan-yuan
    [J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2019, 39 (03) : 917 - 922
  • [7] Deep learning-based garbage image recognition algorithm
    Li, Yuefei
    Liu, Wei
    [J]. APPLIED NANOSCIENCE, 2021, 13 (2) : 1415 - 1424
  • [8] Deep learning-based garbage image recognition algorithm
    Yuefei Li
    Wei Liu
    [J]. Applied Nanoscience, 2023, 13 : 1415 - 1424
  • [9] Research on Spatial Target Classification and Recognition Technology Based on Deep Learning
    Pang, Yujia
    Li, Zhi
    Meng, Bo
    Zhang, Zhimin
    Huang, Longfei
    Huang, Jianbin
    Han, Xu
    Wang, Yin
    Zhu, Xiaohui
    [J]. INTELLIGENT ROBOTICS AND APPLICATIONS, ICIRA 2019, PT IV, 2019, 11743 : 331 - 340
  • [10] Image Recognition for Garbage Classification Based on Transfer Learning and Model Fusion
    Liu, Wei
    Ouyang, Hengjie
    Liu, Qu
    Cai, Sihan
    Wang, Chun
    Xie, Junjie
    Hu, Wei
    [J]. MATHEMATICAL PROBLEMS IN ENGINEERING, 2022, 2022