问题描述
YOLOv5在voc.yaml文件提供了三种方式作为指向训练集和验证集的方式:
# train and val data as
1) directory: path/images/,
2) file: path/images.txt,
or 3) list: [path1/images/, path2/images/]
第一种方式就是直接写文件的目录,第二种方式是写保存了图片文件目录的txt文件,其文件格式如图所示:
笔者在使用第二种方式时会报错,分析主要原因就是标签路径的问题。
解决方案
修改dataset.py文件的img2label_paths
函数,该函数主要的作用是根据imgs图片的路径找到对应labels的路径。
注意这里的对应关系,没有对应就会报错。