Skip to main page content
U.S. flag

An official website of the United States government

Dot gov

The .gov means it’s official.
Federal government websites often end in .gov or .mil. Before sharing sensitive information, make sure you’re on a federal government site.

Https

The site is secure.
The https:// ensures that you are connecting to the official website and that any information you provide is encrypted and transmitted securely.

Access keys NCBI Homepage MyNCBI Homepage Main Content Main Navigation
. 2022 Nov 30;19(23):15987.
doi: 10.3390/ijerph192315987.

An Intelligent Waste-Sorting and Recycling Device Based on Improved EfficientNet

Affiliations

An Intelligent Waste-Sorting and Recycling Device Based on Improved EfficientNet

Zhicheng Feng et al. Int J Environ Res Public Health. .

Abstract

The main source of urban waste is the daily life activities of residents, and the waste sorting of residents' waste is important for promoting economic recycling, reducing labor costs, and protecting the environment. However, most residents are unable to make accurate judgments about the categories of household waste, which severely limits the efficiency of waste sorting. We have designed an intelligent waste bin that enables automatic waste sorting and recycling, avoiding the extensive knowledge required for waste sorting. To ensure that the waste-classification model is high accuracy and works in real time, GECM-EfficientNet is proposed based on EfficientNet by streamlining the mobile inverted bottleneck convolution (MBConv) module, introducing the efficient channel attention (ECA) module and coordinate attention (CA) module, and transfer learning. The accuracy of GECM-EfficientNet reaches 94.54% and 94.23% on the self-built household waste dataset and TrashNet dataset, with parameters of only 1.23 M. The time of one recognition on the intelligent waste bin is only 146 ms, which satisfies the real-time classification requirement. Our method improves the computational efficiency of the waste-classification model and simplifies the hardware requirements, which contributes to the residents' waste classification based on intelligent devices.

Keywords: EfficientNet; artificial intelligence; image classification; sustainable development; waste sorting and recycling.

PubMed Disclaimer

Conflict of interest statement

The authors declare no conflict of interest.

Figures

Figure 1
Figure 1
The intelligent waste bin mechanical modeling and simulation modeling are shown in (a) and (b) respectively, where they were created by SolidWorks. Subfigure (c) shows the physical construction of the intelligent waste bin, which is constructed by the camera, servos, bins, infrared sensor, baffle plate, and paddle plate.
Figure 2
Figure 2
Control circuit. It is built through the Raspberry Pi 4B, camera, servos, and infrared sensor.
Figure 3
Figure 3
MBConv module structure.
Figure 4
Figure 4
Improved EfficientNet structure.
Figure 5
Figure 5
Efficient channel attention module.
Figure 6
Figure 6
Coordinate attention module.
Figure 7
Figure 7
The results of the ablation experiment on the self-built dataset. Subfigure (a) shows the loss curve, where the loss value is calculated on the training dataset. Subfigure (b) shows the accuracy curve, where the accuracy is calculated on the test dataset.
Figure 8
Figure 8
The results of the model performance comparison on the self-built dataset. Subfigure (a) shows the loss curve, where the loss value is calculated on the training dataset. Subfigure (b) shows the accuracy curve, where the accuracy is calculated on the test dataset.
Figure 9
Figure 9
The results of the model performance comparison on the TrashNet dataset. Subfigure (a) shows the loss curve, where the loss value is calculated on the training dataset. Subfigure (b) shows the accuracy curve, where the accuracy is calculated on the test dataset.
Figure 10
Figure 10
Inference time on the Raspberry Pi 4B.
Figure 11
Figure 11
Confusion matrix of GECM-EfficientNet on the test dataset of the self-built dataset. The x-axis and y-axis represent the true and predicted categories of waste respectively, and the different categories are represented by numbers. Subfigure (a) shows the number of predictions for each waste category. Subfigure (b) is obtained by normalizing subfigure (a).
Figure 12
Figure 12
Test results for partial waste images, where the waste type, name and classification accuracy are shown.

References

    1. Tang D., Shi L., Huang X., Zhao Z., Zhou B., Bethel B.J. Influencing Factors on the Household-Waste-Classification Behavior of Urban Residents: A Case Study in Shanghai. Int. J. Environ. Res. Public Health. 2022;19:6528. doi: 10.3390/ijerph19116528. - DOI - PMC - PubMed
    1. Yang Q., Fu L., Liu X., Cheng M. Evaluating the Efficiency of Municipal Solid Waste Management in China. Int. J. Environ. Res. Public Health. 2018;15:2448. doi: 10.3390/ijerph15112448. - DOI - PMC - PubMed
    1. Cheah C.G., Chia W.Y., Lai S.F., Chew K.W., Chia S.R., Show P.L. Innovation designs of industry 4.0 based solid waste management: Machinery and digital circular economy. Environ. Res. 2022;213:113619. doi: 10.1016/j.envres.2022.113619. - DOI - PubMed
    1. Bircanoğlu C., Atay M., Beşer F., Genç Ö., Kızrak M.A. RecycleNet: Intelligent Waste Sorting Using Deep Neural Networks; Proceedings of the 2018 Innovations in Intelligent Systems and Applications (INISTA); Thessaloniki, Greece. 3–5 July 2018; pp. 1–7. - DOI
    1. Vo A.H., Le H.S., Vo M.T., Le T. A Novel Framework for Trash Classification Using Deep Transfer Learning. IEEE Access. 2019;7:178631–178639. doi: 10.1109/ACCESS.2019.2959033. - DOI

LinkOut - more resources