YOLO找不到label的解决方案

之前在训练自己的数据集的时候，使用txt文件（如下图所示）记录分割后的图片内容，一直显示找不到label文件。

其中，train.txt的内容如下

 

找了很多都不能解决问题，于是自己记录一下解决的方案。

解决方案

 1.保证图片和对应的label的txt文件在同一个文件夹下，如下图

2.修改yolo源码中的datasets.py（在utils文件夹下）的img2label_paths函数，修改后函数如下

def img2label_paths(img_paths):
    # Define label paths as a function of image paths
    sa, sb = os.sep + 'images' + os.sep, os.sep + 'labels' + os.sep  # /images/, /labels/ substrings
    return [x.rsplit('.', 1)[0] + '.txt' for x in img_paths]

 然后就不存在问题了，可以正常运行。