1、“yolov5-master\data\coco128-seg.yaml”中设置数据集的目录结构以及分类名称,使用官方数据集的话,需要修改path值,将前面的两个点去掉: 2、“yolov5-master\models\segmentyolov5s-seg.yaml”中设置数据集的分类数量,使用官方数据集的话,直接使用默认配置即可。 2、GPU训练与测试 GPU训练命令如下: PYTHON py...
1)数据集的配置在yolov5目录下的data文件夹下新建一个 wet_waste.yaml文件(可以自定义命名),用来存放训练集和验证集的划分文件(train.txt和val.txt),这两个文件是通过运行voc_label.py代码生成的,然后是目标的类别数目和具体类别列表,wet_waste.yaml内容如下:...