Improving resource recycling based on deep learning

Article type: Research Article

Authors: Xu, Yunjian^a | Guo, Aiyin^{b; *}

Affiliations: [a] School of Intelligent Engineering, Guangdong AIB Polytechnic, Guangzhou, China | [b] School of Internet of Things Application Technology, Guangdong AIB Polytechnic, Guangzhou, China

Correspondence: [*] Corresponding author. E-mail: [email protected].

Abstract: The manual sorting of recyclable garbage has caused several issues such as the wastage of human resources and low resource utilization. To solve this problem, an improved Single Shot Multibox Detector (SSD) deep learning approach has been developed for recyclable garbage detection. To reduce the number of parameters and make the model easier to deploy and apply, a lightweight network called RepVGG has been chosen to replace the VGG16 network in the SSD. Additionally, the auxiliary convolutional layer structure of the SSD has been modified to further reduce the number of parameters. Additionally, the SK module has been integrated to adaptively adjust the size of the receptive field and enhance the detection accuracy. Experimental results of Waste Classification data set from Kaggle website have demonstrated that the improved SSD model has better detection accuracy and real-time performance, with an accuracy of 95.23%, which is 4.33 percentage points higher than the original SSD, and a detection speed of up to 64 FPS. This algorithm can be better applied in industry.

Keywords: Recyclable garbage detection, SSD algorithm, structure re-parameterization, deep learning

DOI: 10.3233/AIS-230124

Journal: Journal of Ambient Intelligence and Smart Environments, vol. Pre-press, no. Pre-press, pp. 1-14, 2024