Garbage detection and classification method based on YoloV5 algorithm

Zhaohao Lv; Huiyan Li; Yeming Liu

doi:10.1117/12.2622439

4 March 2022 Garbage detection and classification method based on YoloV5 algorithm

Zhaohao Lv, Huiyan Li, Yeming Liu

Proceedings Volume 12084, Fourteenth International Conference on Machine Vision (ICMV 2021); 1208403 (2022) https://doi.org/10.1117/12.2622439
Event: Fourteenth International Conference on Machine Vision (ICMV 2021), 2021, Rome, Italy

Abstract

In the face of a wide variety and a large number of production and domestic waste, it is a great challenge for the task of automatic detection and sorting of waste. Based on yolov5 algorithm, this paper proposes a method for rapid detection and classification of garbage, trains the model on taco[1] garbage data set, and extracts the location and feature information of garbage through this network model according to the experimental results. In reality, this model can effectively detect the garbage classified by the data set. After testing, the mAP(Mean Average Percision) value of the model reaches 97.62%, the detection accuracy is 95.49%, and the detection speed reaches 5.52fps. Compared with yolov3 network model, which better complete the task of garbage classification and detection. This network model has the necessary technical conditions for the algorithm of waste sorting robots.

Citation Download Citation

Zhaohao Lv, Huiyan Li, and Yeming Liu "Garbage detection and classification method based on YoloV5 algorithm", Proc. SPIE 12084, Fourteenth International Conference on Machine Vision (ICMV 2021), 1208403 (4 March 2022); https://doi.org/10.1117/12.2622439

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