Sage Journals: Discover world-class research

Abstract

In light of the rapid development of human society, there has been a notable surge in waste production, which has resulted in environmental pollution and degradation. This is a pervasive issue that requires attention. To address the environmental problems caused by waste generation and advance the development of recyclable domestic waste detection, this article proposes waste classification as a solution. Traditional waste sorting methods have proven to be inefficient and prone to errors, hence the need for a more effective approach. A multiobjective recyclable domestic waste detection and classification method based on improved You Only Look Once v5s (YOLOv5s) is proposed in this study. In this study, the network structure is enhanced through the implementation of the Bidirectional Pyramid Network (BiFPN). The coordinate attention mechanism is then incorporated to elevate the accuracy of the model. Additionally, the loss function is refined by adopting the Efficient Intersection Over Union Loss (EIOU_Loss) metric to further optimize network performance. Finally, the introduction of the Ghost convolution module reduces parameter count and significantly improves the real-time detection speed. The waste dataset named Multi-classified Recyclable Domestic Trash Identification Dataset (MULTI-TRASH), which is composed of machine shooting, web crawler, and artificial photography, is used for verification due to its good generalization and representativeness. The mean Average Precision at a threshold of 0.5 (mAP@0.5) value of 94.8% is achieved by the improved model, which is a 30.72% reduction in the number of parameters and a 1.2% improvement in the mAP@0.5 value compared with YOLOv5s. The effectiveness of the proposed algorithm is proved by a comparison with other target detection algorithms. This study aims to provide technical references for the development of a recyclable domestic waste detection system.

Get full access to this article

View all access options for this article.

References

Adedeji

, Wang

. Intelligent waste classification system using deep learning. Proc Manufact, 2019; 35:607–612; doi: 10.1016/j.promfg.2019.05.086

Bochkovskiy

, Wang

C-Y

, Liao

H-YM

. YOLOv4: Optimal speed and accuracy of object detection. arXiv preprint, 2020; doi: 10.48550/arXiv.2004.10934

Carrera

, Piñol

, Mata

, et al. A machine learning based classification models for plastic recycling using different wavelength range spectrums. J Cleaner Prod, 2022; 374:133883; doi: 10.1016/j.jclepro.2022.133883

Chen

, Huang

, Xiao

, et al. Carbon emissions under different domestic waste treatment modes induced by garbage classification: Case study in pilot communities in Shanghai, China. Sci Total Environ, 2020; 717:137193; doi: 10.1016/j.scitotenv.2020.137193

Ding

, Tao

. Trunk-branch ensemble convolutional neural networks for video-based face recognition. IEEE Transact Pattern Anal Machine Intelligence, 2018; 40(4):1002–1014; doi: 10.1109/TPAMI.2017.2700390

Esteva

, Kuprel

, Novoa

, et al. Dermatologist-level classification of skin cancer with deep neural networks. Nature, 2017; 542(7639):115–118; doi: 10.1038/nature21056

Han

, Wang

, Tian

, et al. GhostNet: More features from cheap operations. 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition, CVPR, Seattle, USA; 2020; pp. 1577–1586; doi: 10.1109/CVPR42600.2020.00165

Hou

, Zhou

, Feng

. Coordinate attention for efficient mobile network design. 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition, CVPR,, 2021; pp. 13708–13717; doi: 10.1109/CVPR46437.2021.01350

Howard

, Sandler

, Chen

, et al. Searching for mobileNetV3. 17th IEEE/CVF International Conference on Computer Vision, ICCV, Seoul, Korea (South); 2019; pp. 1314–1324; doi: 10.1109/ICCV.2019.00140

10.

, Shen

, Albanie

, et al. Squeeze-and-Excitation Networks. IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, USA; 2019; p. 8; doi: 10.1109/CVPR.2018.00745

11.

Khan

, Almalaise Alghamdi

, Abushark

, et al. Recycling waste classification using emperor penguin optimizer with deep learning model for bioenergy production. Chemosphere, 2022; 307(Pt 3):136044; doi: 10.1016/j.chemosphere.2022.136044

12.

, Zhang

, Zheng

H-T

, et al. Shufflenet V2: Practical guidelines for efficient cnn architecture design. 15th European Conference on Computer Vision, ECCV, Munich, Germany; 2018; vol. 11218 LNCS; pp. 122–138; doi: 10.1007/978-3-030-01264-9_8

13.

Mao

W-L

, Chen

W-C

, Fathurrahman

HIK

, et al. Deep learning networks for real-time regional domestic waste detection. J Clean Prod, 2022; 344:131096; doi: 10.1016/j.jclepro.2022.131096

14.

Mao

W-L

, Chen

W-C

, Wang

C-T

, et al. Recycling waste classification using optimized convolutional neural network. Resources Conservation Recycl, 2021; 164:105132; doi: 10.1016/j.resconrec.2020.105132

15.

Molchanov

, Tyree

, Karras

, et al. Pruning convolutional neural networks for resource efficient inference. 5th International Conference on Learning Representations, ICLR, Toulon, France; 2017; doi: 10.48550/arXiv.1611.06440

16.

Nowakowski

, Pamula

. Application of deep learning object classifier to improve e-waste collection planning. Waste Manage, 2020; 109:1–9; doi: 10.1016/j.wasman.2020.04.041

17.

Rad

, von Kaenel

, Droux

, et al. A computer vision system to localize and classify wastes on the streets. Comput Vision Syst, 2017; 57:195–204; doi: 10.1007/978-3-319-68345-4_18

18.

Redmon

, Divvala

, Girshick

, et al. You Only Look Once: Unified, Real-Time Object Detection. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, USA; 2016; pp. 779–788; doi: 10.1109/cvpr.2016.91

19.

Redmon

, Farhadi

. YOLO9000: Better, Faster, Stronger. 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Hawaii, USA; 2017; pp. 6517–6525; doi: 10.1109/cvpr.2017.690

20.

Redmon

, Farhadi

. YOLOv3: An Incremental Improvement. arXiv preprint, 2018; doi: 10.48550/arXiv.1804.02767

21.

Sarc

, Curtis

, Kandlbauer

, et al. Digitalisation and intelligent robotics in value chain of circular economy oriented waste management—A review. Waste Manage, 2019; 95:476–492; doi: 10.1016/j.wasman.2019.06.035

22.

Simon

, Rodner

. Neural Activation Constellations: Unsupervised Part Model Discovery with Convolutional Networks. 2015 IEEE International Conference on Computer Vision (ICCV), Santiago, Chile; 2015; pp. 1143–1151; doi: 10.1109/iccv.2015.136

23.

Sozzi

, Cantalamessa

, Cogato

, et al. Automatic bunch detection in white grape varieties using YOLOv3, YOLOv4, and YOLOv5 deep learning algorithms. Agronomy, 2022; 12(2):319; doi: 10.3390/agronomy12020319

24.

Tan

, Pang

, Le

. EfficientDet: Scalable and Efficient Object Detection. 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, USA; 2020; pp. 10778–10787; doi: 10.1109/cvpr42600.2020.01079

25.

, Hoang Son

, Vo

, et al. A novel framework for trash classification using deep transfer learning. IEEE Access, 2019; 7:178631–178639; doi: 10.1109/access.2019.2959033

26.

Woo

, Park

, Lee

J-Y

, et al. CBAM: Convolutional block attention module. 15th European Conference on Computer Vision, ECCV,, 2018; vol. 11211 LNCS; pp. 3–19; doi: 10.1007/978-3-030-01234-2_1

27.

Yang

, Thung

Classification of Trash for Recyclability Status. CS229 Project Report, 2016; p. 3.

28.

Yun

, Han

, Chun

, et al. CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Features. 2019 IEEE/CVF International Conference on Computer Vision (ICCV), Seoul, Korea (South); 2019; pp. 6022–6031; doi: 10.1109/iccv.2019.00612

29.

Zhang

, Yang

, Zhang

, et al. A multi-label waste detection model based on transfer learning. Resources Conserv Recycl, 2022a;181:106235; doi: 10.1016/j.resconrec.2022.106235

30.

Zhang

Y-F

, Ren

, Zhang

, et al. Focal and efficient IOU loss for accurate bounding box regression. Neurocomputing, 2022b;506:146–157; doi: 10.1016/j.neucom.2022.07.042

31.

Zheng

, Wang

, Ren

, et al. Enhancing geometric factors in model learning and inference for object detection and instance segmentation. IEEE Transact Cybernet, 2022; 52(8):8574–8586; doi: 10.1109/TCYB.2021.3095305

A Lightweight Deep Learning Algorithm for Multi-Objective Detection of Recyclable Domestic Waste

Abstract

Get full access to this article

References