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 

机构地区：[1]Faculty of Applied Sciences,Macao Polytechnic University,Rua de Luís Gonzaga Gomes,Macao 999078,China

出　　处：《Visual Computing for Industry,Biomedicine,and Art》2023年第1期279-287,共9页工医艺的可视计算（英文）

摘　　要：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.

关 键 词：Waste recycling Waste classification Knowledge distillation LIGHTWEIGHT Attention 

分 类 号：TP391.7[自动化与计算机技术—计算机应用技术] X70[自动化与计算机技术—计算机科学与技术]