Journal | [J] 计算机系统网络和电信 Volume 5, Issue 8. 2023.
基于 YOLOv5 的垃圾分类识别模型
作者 : 赵 泓宣, 邹 佳欣, 田  朕, 胡旭 栋*, 尼玛 扎西
摘要 / Abstract
为了响应国家推行的垃圾分类,协助公民有效进行垃圾分类丢弃,我们研究了基于图像的垃圾检测与分类模型, 以实现对垃圾的识别和检测。在 GPU 服务器上训练基 于 YOLOv5 s 的垃圾分类模型。然后将训练好的模型 部署到服务器上, 通过微信小程序用户拍照上传照片到服务器,服务器通过模型对图片进行处理并将处理过的图片返回到微信小程序。用户 可以通过照片来判断垃圾是属于哪一类,从而将垃圾进行分类。最终训练出的模型可以识别 44 类的垃圾,同时在识别精度 及响应速度上均有较好的表现。
关键词 / Keywords
YOLOv5s 网络;垃圾分类;目标检测
《中国学术期刊(光盘版)》电子杂志社有限公司 KDN平台基础技术由KBASE 11.0提供