Garbage Classification and Recognition Model based on YOLOv5

Authors

  • Hongxuan Zhao Department of computer science, College of information science and technology, Tibet University, Lhasa 850000, Tibet, China
  • Xudong Hu Department of computer science, College of information science and technology, Tibet University, Lhasa 850000, Tibet, China

Keywords:

YOLOv5s network, Garbage classification, Object detection

Abstract

In response to the garbage classification implemented by the state and to assist citizens in effectively sorting and discarding garbage, we studied the image-based garbage detection and classification model to realize the recognition and detection of garbage. The garbage classification model based on YOLOv5s is trained on the GPU server. Then the trained model is deployed to the server, and the user of wechat applet takes photos and uploads photos to the server. The server processes the images through the model and returns the processed images to the wechat applet. Users can determine the category of garbage through photos, so as to classify the garbage. Finally, the trained model can identify 44 types of garbage, and has good performance in recognition accuracy and response speed.

References

Wang, Z. (2024, August). CausalBench: A Comprehensive Benchmark for Evaluating Causal Reasoning Capabilities of Large Language Models. In Proceedings of the 10th SIGHAN Workshop on Chinese Language Processing (SIGHAN-10) (pp. 143-151).

Wang, Z., Zhu, Y., He, S., Yan, H., & Zhu, Z. (2024). LLM for Sentiment Analysis in E-commerce: A Deep Dive into Customer Feedback. Applied Science and Engineering Journal for Advanced Research, 3(4), 8-13.

Yuan, B. (2024). Design of an Intelligent Dialogue System Based on Natural Language Processing. Journal of Theory and Practice of Engineering Science, 4(01), 72-78.

Kim, K. , Kim, K. , Jeong, S. , & Li, M. E. . (2023). Application of YOLO v5 and v8 for Recognition of Safety Risk Factors at Construction Sites.

Li, L. , Shuai, Z. , Hu, J. , & Zhang, Y. . (2022). Classification of Tropical Cyclone Intensity Based on Deep Learning and YOLO V5. International Conference on Artificial Intelligence and Security. Springer, Cham.

Li, P. , & Zhu, C. . (2024). Ro-yolov5: one new detector for impurity in wheat based on circular smooth label. Crop Protection, 184.

Lyu, H., Wang, Z., & Babakhani, A. (2020). A UHF/UWB hybrid RFID tag with a 51-m energy-harvesting sensitivity for remote vital-sign monitoring. IEEE transactions on microwave theory and techniques, 68(11), 4886-4895.

Yao, J., & Yuan, B. (2024). Research on the Application and Optimization Strategies of Deep Learning in Large Language Models. Journal of Theory and Practice of Engineering Science, 4(05), 88-94.

Wang, Z., Zhu, Y., Li, Z., Wang, Z., Qin, H., & Liu, X. (2024). Graph neural network recommendation system for football formation. Applied Science and Biotechnology Journal for Advanced Research, 3(3), 33-39.

Zhang, X. , Yan, M. , Zhu, D. , & Guan, Y. . (2022). Marine ship detection and classification based on YOLOv5 model.

Sun, T. , Liu, B. , Zheng, R. , & Peng, Z. . (2023). Research on multi-target recognition and classification strategy based on yolo v5 framework. Springer, Singapore.

Kim, J. H. , Kim, N. , Park, Y. W. , & Won, C. S. . (2022). Object Detection and Classification Based on YOLO-V5 with Improved Maritime Dataset.

Liu, Z. , Liu, H. , & Sun, Y. . Detection and Classification of Malaria Parasite Based on Improved YOLOv5 Model. 2023 16th International Congress on Image and Signal Processing, BioMedical Engineering and Informatics (CISP-BMEI). IEEE.

Xu, X., Yuan, B., Song, T., & Li, S. (2023, November). Curriculum recommendations using transformer base model with infonce loss and language switching method. In 2023 5th International Conference on Artificial Intelligence and Computer Applications (ICAICA) (pp. 389-393). IEEE.

Jabed, M. R. , & Shamsuzzaman, M. . (2022). Yolobin: non-decomposable garbage identification and classification based on yolov7. 10(10), 104-121.

Wang, J. , Zhan, X. , Yang, Z. , & Li, Y. . Enhanced and improved garbage identification and classification of YOLOV5 based on data. 2024 5th International Conference on Computer Vision, Image and Deep Learning (CVIDL). IEEE.

Zheng, Y., Zhou, G., & Lu, B. (2023). Rebar Cross-section Detection Based on Improved YOLOv5s Algorithm. Innovation & Technology Advances, 1(1), 1–6. https://doi.org/10.61187/ita.v1i1.1

Xu, Z. , Liu, K. , & Gu, X. G. K. . (2023). Image recognition model of pipeline magnetic flux leakage detection based on deep learning. Corrosion Reviews, 41(6), 689-701.

Zhao, X., Zhang, L., & Hu, Z. (2023). Smart warehouse track identification based on Res2Net-YOLACT+HSV. Innovation & Technology Advances, 1(1), 7–11. https://doi.org/10.61187/ita.v1i1.2

Zhu, Z., Wang, Z., Wu, Z., Zhang, Y., & Bo, S. (2024). Adversarial for Sequential Recommendation Walking in the Multi-Latent Space. Applied Science and Biotechnology Journal for Advanced Research, 3(4), 1-9.

Yuan, B., & Song, T. (2023, November). Structural Resilience and Connectivity of the IPv6 Internet: An AS-level Topology Examination. In Proceedings of the 4th International Conference on Artificial Intelligence and Computer Engineering (pp. 853-856).

Wang, C. , Zhang, Y. , Zhou, Y. , Sun, S. , Zhang, H. , & Wang, Y. . (2023). Automatic detection of indoor occupancy based on improved yolov5 model. Neural computing & applications.

Yuan, B., Song, T., & Yao, J. (2024, January). Identification of important nodes in the information propagation network based on the artificial intelligence method. In 2024 4th International Conference on Consumer Electronics and Computer Engineering (ICCECE) (pp. 11-14). IEEE.

Wang, R. , Zhang, Z. F. , & Yang, BenXi, Hai-QiZhai, Yu-ShengZhang, Rui-LiangGeng, Li-JieChen, Zhi-YongYang, Kun. (2023). Detection and classification of cotton foreign fibers based on polarization imaging and improved yolov5. Sensors (Basel, Switzerland), 23.

Lin, Z., Wang, Z., Zhu, Y., Li, Z., & Qin, H. (2024). Text Sentiment Detection and Classification Based on Integrated Learning Algorithm. Applied Science and Engineering Journal for Advanced Research, 3(3), 27-33.

Zhang, X. , Yan, M. , Zhu, D. , & Guan, Y. . (2022). Marine ship detection and classification based on yolov5 model. Journal of Physics: Conference Series, 2181(1), 012025-.

Yao, J., & Yuan, B. (2024). Optimization Strategies for Deep Learning Models in Natural Language Processing. Journal of Theory and Practice of Engineering Science, 4(05), 80-87.

Lv, Z. , Li, H. , & Liu, Y. . (2022). Garbage detection and classification method based on yolov5 algorithm.

Xu, J., Jiang, Y., Yuan, B., Li, S., & Song, T. (2023, November). Automated Scoring of Clinical Patient Notes using Advanced NLP and Pseudo Labeling. In 2023 5th International Conference on Artificial Intelligence and Computer Applications (ICAICA) (pp. 384-388). IEEE.

Downloads

Published

2024-08-15

How to Cite

Zhao, H., & Hu, X. (2024). Garbage Classification and Recognition Model based on YOLOv5. Journal of Theory and Practice in Engineering and Technology, 1(2), 8–13. Retrieved from https://woodyinternational.com/index.php/jtpet/article/view/43