我被困在這里,我是一個(gè)新手,我認(rèn)為問(wèn)題在于將數(shù)據(jù)集轉(zhuǎn)換為 yolov4 格式,因?yàn)槲沂褂昧诉@段代碼: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)我認(rèn)為這個(gè)程序是錯(cuò)誤的任何幫助將不勝感激。
2 回答

慕哥6287543
TA貢獻(xiàn)1831條經(jīng)驗(yàn) 獲得超10個(gè)贊
我還不知道如何解決,但我知道是什么原因造成的。
我嘗試使用 6 通道圖像進(jìn)行訓(xùn)練,但 Yolo 內(nèi)部使用 OpenCV,目前無(wú)法讀取超過(guò) 3 通道的圖像。
如果不是這種情況,則它必須是以下之一
檢查train.txt和obj.data文件是否配置正確
檢查您是否可以在 python > opencv 中打開(kāi)并從拋出錯(cuò)誤的數(shù)據(jù)集中讀取文件。
添加回答
舉報(bào)
0/150
提交
取消