我们在使用yolov5训练数据集的时候,如果数据集是VOC或是coco分布和命名格式的不需要修改,而其他格式的数据集需要在datasets.py文件中修改。
在datasets.py文件中找到函数:"def img2label_paths(img_paths):"
我的是yolov5-5.0,在341行
将函数里面的images和labels换成自己数据集里面图片和标签所在文件夹的名字即可。
我们在使用yolov5训练数据集的时候,如果数据集是VOC或是coco分布和命名格式的不需要修改,而其他格式的数据集需要在datasets.py文件中修改。
在datasets.py文件中找到函数:"def img2label_paths(img_paths):"
我的是yolov5-5.0,在341行
将函数里面的images和labels换成自己数据集里面图片和标签所在文件夹的名字即可。