基于 YOLOv5 的垃圾分类识别模型

赵 泓宣, 邹 佳欣, 田 　朕, 胡旭 栋*, 尼玛 扎西
西藏大学信息科学技术学院计算机系

摘要

为了响应国家推行的垃圾分类，协助公民有效进行垃圾分类丢弃，我们研究了基于图像的垃圾检测与分类模型，
以实现对垃圾的识别和检测。在 GPU 服务器上训练基 于 YOLOv5 s 的垃圾分类模型。然后将训练好的模型 部署到服务器上，
通过微信小程序用户拍照上传照片到服务器，服务器通过模型对图片进行处理并将处理过的图片返回到微信小程序。用户
可以通过照片来判断垃圾是属于哪一类，从而将垃圾进行分类。最终训练出的模型可以识别 44 类的垃圾，同时在识别精度
及响应速度上均有较好的表现。

关键词

YOLOv5s 网络；垃圾分类；目标检测

参考

[1] 王莉 , 何牧天 , 徐硕等 . 基于 YOLOv5s 网络的垃圾

分 类 和 检 测 [J]. 包 装 工 程 ,2021,42(08):50-56.DOI:10.19554/

j.cnki.1001-3563.2021.08.007.

[2] 刘 梦 杰 . 基 于 深 度 学 习 的 可 回 收 垃 圾 检 测 算

法 研 究 [D]. 沈 阳 工 业 大 学 ,2022.DOI:10.27322/d.cnki.

gsgyu.2022.000120.

[3] 涂成凤 , 易安林 , 姚涛等 . 轻量化 YOLOv5n 的高精

度垃圾检测算法 [J]. 计算机工程与应用 ,2023,59(10):187-195.

[4]Zhaohao L,Huiyan L,Yeming L. Garbage detection and

classification method based on YoloV5 algorithm[P]. Tian jin

Univ. of Technology and Education (China),2022.

[5] 谈 世 磊 ; 别 雄 波 ; 卢 功 林 ; 谈 小 虎 . 基 于

YOLOv5 网 络 模 型 的 人 员 口 罩 佩 戴 实 时 检 测 [J]. 激 光 杂

志 ,2021,42(02):147-150.DOI:10.14016/j.cnki.jgzz.2021.02.147

[6] 王 莉 ; 何 牧 天 ; 徐 硕 ; 袁 天 ; 赵 天 翊 ; 刘 建

飞 . 基 于 YOLOv5s 网 络 的 垃 圾 分 类 和 检 测 [J]. 包 装 工

程 ,2021,42(08):50-56.DOI:10.19554/j.cnki.1001-3563.2021.

08.007

[7] 邱天衡 ; 王玲 ; 王鹏 ; 白燕娥 . 基于改进 YOLOv5 的

目标检测算法研究 [J]. 计算机工程与应用 ,2022,58(13):63-73.

[8] 赵婉月 . 基于 YOLOv5 的目标检测算法研究 [D]. 西安

电子科技大学 ,2021.DOI:10.27389/d.cnki.gxadu.2021.002918.