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

Question

我被困在这里，我是一个新手，我认为问题在于将数据集转换为 yolov4 格式，因为我使用了这段代码：

import os

import random

imgspath = '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 necessary

trainlen = 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 通道的图像。如果不是这种情况，则它必须是以下之一检查train.txt和obj.data文件是否配置正确检查您是否可以在 python > opencv 中打开并从抛出错误的数据集中读取文件。

Answer 2

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