A lightweight YOLOv5 garbage detection and classification method

Mei Huang; Yongxin Chang; Liangbao Zhang; Shuaifeng Jiao

doi:10.1117/12.2653673

18 November 2022 A lightweight YOLOv5 garbage detection and classification method

Mei Huang, Yongxin Chang, Liangbao Zhang, Shuaifeng Jiao

Proceedings Volume 12473, Second International Conference on Optics and Communication Technology (ICOCT 2022); 124730W (2022) https://doi.org/10.1117/12.2653673
Event: Second International Conference on Optics and Communication Technology (ICOCT 2022), 2022, Hefei, China

Abstract

Aiming at the problems of unclear, difficult, and inefficient classification of traditional manual waste, and the difficulty of deploying large existing garbage classification network models, a lightweight garbage detection and classification network S-YOLOv5 is designed based on YOLOv5s. First, a garbage dataset containing 18 types of common household garbage is constructed and labeled according to the principles of garbage classification; secondly, a module combining shufflenetv2 and CoordAttention was introduced to replace the YOLOv5s backbone network, and the ReLU activation function in the shufflenet module was substituted by FReLU; finally the PANet structure was replaced by the BiFPN structure, so as to reduce the model complexity and achieve lightweight while maintaining a high mAP. The experimental results show that the size of S-YOLOv5 is only 2.6MB, which is about 1/6 of the original network size, and the mAP is 80.2%. The size of the proposed network is reduced while maintaining high accuracy, making it more suitable for deployment in smart devices.

Citation Download Citation

Mei Huang, Yongxin Chang, Liangbao Zhang, and Shuaifeng Jiao "A lightweight YOLOv5 garbage detection and classification method", Proc. SPIE 12473, Second International Conference on Optics and Communication Technology (ICOCT 2022), 124730W (18 November 2022); https://doi.org/10.1117/12.2653673

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available