Focus-RCNet: a lightweight recyclable waste classification algorithm based on focus and knowledge distillation

Dashun Zheng, Rongsheng Wang, Yaofei Duan, Patrick Cheong Iao Pang, Tao Tan

研究成果: Article同行評審

1 引文 斯高帕斯(Scopus)


Waste pollution is a significant environmental problem worldwide. With the continuous improvement in the living standards of the population and increasing richness of the consumption structure, the amount of domestic waste generated has increased dramatically, and there is an urgent need for further treatment. The rapid development of artificial intelligence has provided an effective solution for automated waste classification. However, the high computational power and complexity of algorithms make convolutional neural networks unsuitable for real-time embedded applications. In this paper, we propose a lightweight network architecture called Focus-RCNet, designed with reference to the sandglass structure of MobileNetV2, which uses deeply separable convolution to extract features from images. The Focus module is introduced to the field of recyclable waste image classification to reduce the dimensionality of features while retaining relevant information. To make the model focus more on waste image features while keeping the number of parameters small, we introduce the SimAM attention mechanism. In addition, knowledge distillation was used to further compress the number of parameters in the model. By training and testing on the TrashNet dataset, the Focus-RCNet model not only achieved an accuracy of 92 % but also showed high deployment mobility.

期刊Visual Computing for Industry, Biomedicine, and Art
出版狀態Published - 12月 2023


深入研究「Focus-RCNet: a lightweight recyclable waste classification algorithm based on focus and knowledge distillation」主題。共同形成了獨特的指紋。