yolov4自定义物体检测在训练过程中出错

Question

我被困在这里，我是一个新手，我认为问题在于将数据集转换为 yolov4 格式，因为我使用了这段代码：import osimport randomimgspath = 'C:/yolo_v4/yolo_v4_mask_detection/darknet/build/darknet/x64/data/obj'path = 'data/obj/'images = []for i in os.listdir(imgspath):    temp = path+i    images.append(temp)# train and test split... adjust it if necessarytrainlen = round(len(images)*.80)testlen = round(len(images)*.20)#print('total, train, test dataset size -',trainlen+testlen,trainlen,testlen)random.shuffle(images)test = images[:testlen]train = images[testlen:]with open('train.txt', 'w') as f:    for item in train:        f.write("%s\n" % item)with open('test.txt', 'w') as f:    for item in test:        f.write("%s\n" % item)我认为这个程序是错误的任何帮助将不胜感激。

Answer 1

我还不知道如何解决，但我知道是什么原因造成的。

我尝试使用 6 通道图像进行训练，但 Yolo 内部使用 OpenCV，目前无法读取超过 3 通道的图像。

如果不是这种情况，则它必须是以下之一

1. 检查train.txt和obj.data文件是否配置正确

2. 检查您是否可以在 python > opencv 中打开并从抛出错误的数据集中读取文件。

Answer 2

是你文件路径的问题，检查一下。