介绍 (Introduction)
该项目主要使用YOLOv5模型实现垃圾对象的目标检测任务:
-
所使用的垃圾对象数据集为TACO数据集
-
本项目主要在官方提供的YOLOv5项目的基础上进行一些小的修改
该项目的使用教程,请参见"_README.ipynb"文件
[注意]
-
由于官方YOLOv5模型需要使用的Python版本需要大于等于3.7,本平台的Python版本不支持相关代码的运行。因此,需要使用该模型的研究者请下载项目后在本地运行。
-
因此,该项目的"使用"部分仅提供结果的展示,可以展示的图片位于"源码"部分inference/images文件夹下,图片输入名称不带扩展名(如".jpg")
[补充]
项目中的数据集可能已经失效,如有需要可以使用以下百度网盘链接进行下载
链接:https://pan.baidu.com/s/1F_psRJ5dvgpEYujpoU6YrA
提取码:e36z