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

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.