Recyclable Garbage Detection Method of Improved SSD

doi:10.3778/j.issn.1002-8331.2105-0443

Abstract

Abstract: Nowadays, the recycling and utilization of recyclable garbage in our country mainly relies on manual sorting, which has caused problems such as waste of human resources and low resource utilization. To improve the recycling of resources, a recyclable garbage detector based on improved single shot multibox detector（SSD） algorithm is proposed. In order to solve the problem that the model parameters are too large to deploy, a new feature extraction backbone is selected, a lightweight network RepVGG is introduced to replace the VGG16 network in SSD, and structural re-parameterization method is used to significantly reduce the parameters and computation. It modifies the auxiliary convolution layer structure of SSD to further reduce parameters. For the large change of dataset size, SK module is introduced to adjust the size of the receptive field adaptively and improve the detection accuracy. The experimental results show that the improved SSD model has better detection accuracy and real-time performance in the detection task of recyclable garbage, the accuracy is 95.23%, which is 4.33?percentage points higher than the original SSD, and the detection speed can reach 64?FPS, so the algorithm can be better applied in industry.

Key words: recyclable garbage detection, single shot multibox detector（SSD）, structural re-parameterization, deep learning

摘要： 目前我国可回收垃圾的回收利用主要依靠人工分拣的方式，造成了人力资源浪费、资源利用率低等问题。为提高资源的回收利用，提出了一个基于改进single shot multibox detector（SSD）算法的可回收垃圾检测器。针对模型参数量大，难以部署应用的问题，选用新的主干特征提取网络，引入轻量化的网络RepVGG替换SSD中的VGG16网络，同时采用结构重参数化的方式大幅减少参数量和计算量。修改SSD的辅助卷积层结构，进一步减少参数量。针对数据集尺寸变化大的问题，引入SK模块，自适应调整感受野的尺寸，提高检测精度。实验结果表明，改进的SSD模型在可回收垃圾检测任务上具有较好的检测精度和实时性，精度为95.23%，相较于原始SSD提升了4.33个百分点，检测速度可以达到64?FPS，因此该算法可以更好地应用于工业。

关键词: 可回收垃圾检测, SSD算法, 结构重参数化, 深度学习

GENG Liting, Alifu·Kuerban, Minawaer·Abula, DING Pei, JIANG Runxi. Recyclable Garbage Detection Method of Improved SSD[J]. Computer Engineering and Applications, 2022, 58(23): 293-299.

耿丽婷, 阿里甫·库尔班, 米娜瓦尔·阿不拉, 丁培, 蒋润熙. 改进SSD的可回收垃圾检测方法[J]. 计算机工程与应用, 2022, 58(23): 293-299.

References

[1] 上海市生活垃圾管理条例[J].上海预防医学，2019，31（8）：669.
Regulations of Shanghai Municipality on domestic waste management[J].Shanghai Journal of Preventive Medicine，2019，31（8）：669.
[2] ZOU Z，SHI Z，GUO Y，et al.Object detection in 20 years：a survey[J].arXiv：1905.05055，2019.
[3] GIRSHICK R，DONAHUE J，DARRELL T，et al.Rich feature hierarchies for accurate object detection and semantic segmentation[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2014：580-587.
[4] HE K M，ZHANG X Y，REN S Q，et al.Spatial pyramid pooling in deep convolutional networks for visual recognition[J].IEEE Transactions on Pattern Analysis and Machine Intelligence，2015，37（9）：1904-1916.
[5] GIRSHICK R.Fast R-CNN[C]//Proceedings of the IEEE International Conference on Computer Vision，2015：1440-1448.
[6] REN S，HE K，GIRSHICK R，et al.Faster R-CNN：towards real-time object detection with region proposal networks[J].IEEE Transactions on Pattern Analysis & Machine Intelligence，2017，39（6）：1137-1149.
[7] LIU W，ANGUELOV D，ERHAN D，et al.SSD：single shot multibox detector[C]//European Conference on Computer Vision.Cham：Springer，2016：21-37.
[8] BOCHKOVSKIY A，WANG C Y，LIAO H.YOLOv4：optimal speed and accuracy of object detection[J].arXiv：2004.10934，2020.
[9] REDMON J，FARHADI A.YOLOv3：an incremental improvement[J].arXiv：1804.02767，2018.
[10] LIU Y，GE Z，LV G，et al.Research on automatic garbage detection system based on deep learning and narrowband internet of things[J].Journal of Physics：Conference Series，2018，1069：012032.
[11] HOWARD A G，ZHU M，CHEN B，et al.MobileNets：efficient convolutional neural networks for mobile vision applications[J].arXiv：1704.04861，2017.
[12] 刘恩乾.基于深度学习的生活垃圾分类和检测[D].太原：山西大学，2020.
LIU E Q.Classification and detection of domestic waste based on deep learning[D].Taiyuan：Shanxi University，2020.
[13] 宁凯，张东波，印峰，等.基于视觉感知的智能扫地机器人的垃圾检测与分类[J].中国图象图形学报，2019，24（8）：1358-1368.
NING K，ZHANG D B，YIN F，et al.Garbage detection and classification of intelligent sweeping robot based on visual perception[J].Journal of Image and Graphics，2019，24（8）：1358-1368.
[14] CUI W，ZHANG W，GREEN J，et al.YOLOv3-darknet with adaptive clustering anchor box for garbage detection in intelligent sanitation[C]//Proceedings of the 2019 3rd International Conference on Electronic Information Technology and Computer Engineering，2019.
[15] SIMONYAN K，ZISSERMAN A.Very deep convolutional networks for large-scale image recognition[J].arXiv：1409.1556，2014.
[16] DING X，ZHANG X，MA N，et al.RepVGG：making VGG-style ConvNets great again[J].arXiv：2101.03697，2021.
[17] DING X，GUO Y，DING G，et al.ACNet：strengthening the kernel skeletons for powerful CNN via asymmetric convolution blocks[C]//Proceedings of the IEEE/CVF International Conference on Computer Vision，2019：1911-1920.
[18] LI X，WANG W，HU X，et al.Selective kernel networks[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition，2019：510-519.
[19] JIE H，LI S，GANG S.Squeeze-and-excitation networks[C]//Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition（CVPR），2018：7132-7141.
[20] SZEGEDY C，LIU W，JIA Y，et al.Going deeper with convolutions[C]//2015 IEEE Conference on Computer Vision and Pattern Recognition，2015.
[21] LIU S，HUANG D，WANG Y.Receptive field block net for accurate and fast object detection[J].arXiv：1711.07767，2017.