Paper
27 June 2023 Garbage classification model integrating attention mechanism
Zhen Ye, Kangdi Yin, Lin Bai
Author Affiliations +
Proceedings Volume 12705, Fourteenth International Conference on Graphics and Image Processing (ICGIP 2022); 127051C (2023) https://doi.org/10.1117/12.2680654
Event: Fourteenth International Conference on Graphics and Image Processing (ICGIP 2022), 2022, Nanjing, China
Abstract
In this paper, we propose a garbage classification model that integrates the attention mechanism and multiple network optimization methods. First, we construct a four-category primary network for recyclable garbage, kitchen garbage, hazardous garbage, and other garbage. And then, four secondary networks are constructed to map the above 4 primary categories to 40 secondary classes. Both the primary and secondary networks take Resnet101 as the main backbone network and integrate attention mechanism, Focal loss function, and warm-up learning rate. The experimental results prove that the proposed model has a high classification performance for the HUAWEI cloud garbage classification dataset.
© (2023) COPYRIGHT Society of Photo-Optical Instrumentation Engineers (SPIE). Downloading of the abstract is permitted for personal use only.
Zhen Ye, Kangdi Yin, and Lin Bai "Garbage classification model integrating attention mechanism", Proc. SPIE 12705, Fourteenth International Conference on Graphics and Image Processing (ICGIP 2022), 127051C (27 June 2023); https://doi.org/10.1117/12.2680654
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Deep learning

Feature extraction

Image classification

Back to Top