【目标检测】YOLOv5在voc.yaml使用txt作为目录无法训练的解决方案

问题描述

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的路径。

注意这里的对应关系，没有对应就会报错。