塑料科技2024，Vol.52Issue(11)：140-146,7.DOI:10.15925/j.cnki.issn1005-3360.2024.11.027
基于深度学习的废旧塑料瓶颜色高效识别方法研究
Research on Efficient Color Recognition Method for Waste Plastic Bottles Based on Deep Learning
摘要
Abstract
To address the value difference between different colors in the recycling process of waste plastic bottles,the study provides an efficient color recognition approach for waste plastic bottles based on a deep learning-upgraded YOLOv8n model to handle the problem of color recognition and sorting.Adding a Bidirectional Feature Pyramid Network(BiFPN)to the neck network for multi-scale feature fusion to improve the model's generalizability.The decoupled head structure of the head network uses only one Conv2d module for both branches,and a reparameterized convolution RepConv module is added at the front end of the branch to reduce computational complexity and improve training accuracy.Replace the CIOU loss function with the WIOU v3 loss function to improve the model's detection accuracy.Comparative trials demonstrate that the model suggested in the paper is superior to typical object detection models.The results showed that the model in the article had a 44.8%lower parameter count,a 34.6%lower computational complexity,and a 2.7%higher mean average precision at 50%IOU(mAP50)than the original model.When identifying the colors of waste plastic bottles,the frames per second(FPS)can reach 66,which is 40.4%faster than the original model.The colors of waste plastic bottles may now be detected in real time and with high accuracy.关键词
YOLOv8n/废旧塑料瓶/分类识别/目标检测Key words
YOLOv8n/Waste plastic bottles/Classification recognition/Object detection分类
信息技术与安全科学引用本文复制引用
谢世龙,吴虎,毛文杰,初宪龙,杨先海..基于深度学习的废旧塑料瓶颜色高效识别方法研究[J].塑料科技,2024,52(11):140-146,7.基金项目
国家自然科学基金(52075306) （52075306）
