GARBAGE DETECTION BASED ON LGD-YOLO HIGH PRECISION LIGHTWEIGHT OBJECT DETECTION NETWORK

XIAO Lizhong; HU Fan

doi:10.13205/j.hjgc.202406020

Volume 42 Issue 6

Jun. 2024

Turn off MathJax

Article Contents

Abstract

References

Article Navigation > ENVIRONMENTAL ENGINEERING > 2024 > 42(6): 169-177

XIAO Lizhong, HU Fan. GARBAGE DETECTION BASED ON LGD-YOLO HIGH PRECISION LIGHTWEIGHT OBJECT DETECTION NETWORK[J]. ENVIRONMENTAL ENGINEERING , 2024, 42(6): 169-177. doi: 10.13205/j.hjgc.202406020

Citation:

XIAO Lizhong, HU Fan. GARBAGE DETECTION BASED ON LGD-YOLO HIGH PRECISION LIGHTWEIGHT OBJECT DETECTION NETWORK[J]. ENVIRONMENTAL ENGINEERING , 2024, 42(6): 169-177. doi: 10.13205/j.hjgc.202406020

XIAO Lizhong, HU Fan. GARBAGE DETECTION BASED ON LGD-YOLO HIGH PRECISION LIGHTWEIGHT OBJECT DETECTION NETWORK[J]. ENVIRONMENTAL ENGINEERING , 2024, 42(6): 169-177. doi: 10.13205/j.hjgc.202406020

Citation:

XIAO Lizhong, HU Fan. GARBAGE DETECTION BASED ON LGD-YOLO HIGH PRECISION LIGHTWEIGHT OBJECT DETECTION NETWORK[J]. ENVIRONMENTAL ENGINEERING , 2024, 42(6): 169-177. doi: 10.13205/j.hjgc.202406020

PDF( 11272 KB)

GARBAGE DETECTION BASED ON LGD-YOLO HIGH PRECISION LIGHTWEIGHT OBJECT DETECTION NETWORK

doi: 10.13205/j.hjgc.202406020

XIAO Lizhong,
HU Fan^,

School of Computer Science and Information Engineering, Shanghai Institute of Technology, Shanghai 201418, China

Received Date: 2023-05-23
Available Online: 2024-07-11

Abstract

Abstract

At present, with the rapid development of society and the rapid growth of the urban population, garbage pollution has become increasingly prominent, and garbage classification is imperative. Manual processing has the problems of heavy tasks and low efficiency, and some automated classification methods have low detection accuracy and slow speed. To improve the accuracy of garbage detection in complex scenes, lighten the structure, and make it easier to deploy, an improved YOLO v5s garbage detection model, Lightweight Garbage Detection YOLO (LGD-YOLO) was proposed, which integrated a lightweight convolution module, attention mechanism, and multiple receptive field modules. First, Ghost convolution and Slim Neck module including GSConv were introduced into the network structure to make the model lighter. Secondly, the coordinate attention mechanism was embedded to focus on important information to improve the detection accuracy. Finally, the multi-receptive field module was introduced to improve the multi-scale detection capability of the model and avoid missing detection of small target objects. The data set containing trash garbage images in different environments was tested and verified. The experimental results showed that the parameters and calculation amount of the improved model were 5.77 M and 9.2 GFLOPs, respectively, which were 22.4% and 56.4% less than the original model. The single image detection speed was 26.5 ms, meeting the real-time requirements of garbage detection. In addition, the improved algorithm has good detection accuracy, with mAP_0.5 and mAP_0.5∶0.95 reaching 96.20% and 77.77%, respectively, which was superior to the current popular target detection algorithm.
- garbage classification,
- object detection,
- lightweight network,
- coordinate attention,
- multiscale detection

FullText(HTML)

References(21)

References

[1]	李金玉,陈晓雷,张爱华,等. 基于深度学习的垃圾分类方法综述[J]. 计算机工程,2022,48(2):1-9.
[2]	LIU W, ANGUELOY D, ERHAN D, et al. SSD: single shot multibox detector[C]//European Conference on Computer Vision, 2016:21-37.
[3]	REDMON J,DIVVALA S,GIRSHICK R, et al. You only look once: unified, real-time object detection[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2016:779-788.
[4]	REDMON J,FARHADI A.YOLOv3: An Incremental Improvement[D].Washington: University of Washington,2018.
[5]	GIRSHICK R, DONAHUE J, DARRELL T, et al. Region-based convolutional networks for accurate object detection and segmentation[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2016, 38(1): 142-158.
[6]	GIRSHICK R. Fast R-CNN[C]//Proceedings of the IEEE International Conference on Computer Vision. 2015:1440-1448.
[7]	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.
[8]	王莉,何牧天,徐硕,等.基于YOLOv5s网络的垃圾分类和检测[J].包装工程,2021,42(8):50-56.
[9]	马雯,于炯.基于改进Faster R-CNN的垃圾检测与分类方法[J].计算机工程,2021,47(8):294-300.
[10]	罗安能,万海斌.基于改进YOLOv5s的可回收垃圾的检测算法[J/OL].激光与光电子学进展:1-15[2022-11-15 ].
[11]	HAN K, WANG Y, TIAN Q, et al. GhostNet: More features from cheap operations[C]//2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). IEEE, 2020.
[12]	LI H L, LI J, WEI H B, et al. Slim-neck by GSConv: a better design paradigm of detector architectures for autonomous vehicles[C]//ArXiv abs/2206.02424 (2022): n. pag.
[13]	JIE H,LI S,GANG S, et al. Squeeze-and-excitation networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017,99:.
[14]	HOU Q, ZHOU D, FENG J. Coordinate attention for efficient mobile network design[C]//2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR),2021: 13708-13717.
[15]	LIU S T, HUANG D, WANG Y H. Receptive field block net for accurate and fast object detection[C]//European Conference on Computer Vision,2018: 404-419.
[16]	YANG M, THUNG G. Classification of trash for recyclability status[EB/OL]. Stanford University Machine Learning Project Report, CS229, 2016[2021-01-25].
[17]	PROENÇA PEDRO F, PEDRO Simões. TACO: Trash Annotations in Context for Litter Detection[C]//ArXiv abs/2003.06975 (2020): n. pag.
[18]	WOO S, PARK J, LEE J Y, et al. Cbam: convolutional block attention module[C]//Proceedings of the European Conference on Computer Vision (ECCV), 2018: 3-19.
[19]	王一田,唐开强,留沧海,等.基于YOLO v3的地面垃圾检测与清洁度评定方法[J].传感器与微系统,2022(4):129-133.
[20]	袁红春,臧天祺.基于注意力机制Ghost-YOLOv5的水下垃圾目标检测[J].环境工程,2023,41(7):214-221.
[21]	JIANG X, HU H, QIN Y, et al. A real-time rural domestic garbage detection algorithm with an improved YOLOv5s network model[C]//Scientific Reports,2022,12: n. pag.