本项目展示了如何使用YOLOv5算法进行垃圾分类目标检测。通过利用已标注的目标检测数据集进行训练,我们可以准确地识别居民生活垃圾的类别,并在图像中指示其位置。本项目使用基于PyTorch的ultralytics/yolov5实现,适用于Windows系统。具体的项目流程包括数据集准备和格式转换、探索性数据分析、环境安装、YOLOv5代码修改(以支持中文标签)、训练集和测试集的自动划分、配置文件的修改、Weights&Biases训练可视化工具的准备、网络模型的训练以及测试和性能统计分析。
YOLOv5目标检测在垃圾分类中的应用实例
文件列表
基于YOLOv5实现垃圾分类目标检测
(预估有个2000文件)
.gitattributes
75B
events.out.tfevents.1630724836.LAPTOP-2PUAK7IN.29844.0
37KB
2c57f9a2fbb0d4e2fdc8968bce13cc03.jpeg
92KB
2f5c39d5fe55bf7926444c6e9e177a96.jpeg
185KB
8c1cedcc5f1c7e28619b160d5c46504d.jpeg
40KB
c0637f2d09360f7980f498ce1bec0b61.jpeg
42KB
570ce97066145f9179e56d616f7ee903.jpeg
67KB
662d79ab3f8568e95e9d2835f416d0a5.jpeg
106KB
Dockerfile
821B
yolov5-trash(1).iml
495B
暂无评论