无法打开标签文件。(只有使用MSCOCO时才能正常使用) YoloV4

Question

我正在处理YoloV4模型对象检测。我正在尝试训练自定义数据集，但我经常得到以下错误行：

Can't open label file. (This can be normal only if you use MSCOCO): data/obj/13_PNG.rf.c87d3ef90086ec0d21254a8a7c97147a.txt 
Can't open label file. (This can be normal only if you use MSCOCO): data/obj/13_PNG.rf.c87d3ef90086ec0d21254a8a7c97147a.txt 
Can't open label file. (This can be normal only if you use MSCOCO): data/obj/13_PNG.rf.c87d3ef90086ec0d21254a8a7c97147a.txt 

培训文件路径似乎不匹配，但我不知道如何解决这个问题。我挣扎了好几个小时。

我试图使用的命令运行培训过程：

!./darknet detector train data/obj.data cfg/custom-yolov4-detector.cfg yolov4.conv.137 -dont_show

以及火车的文件和目录：

%cd /content/darknet/                                   
%cp {dataset.location}/train/_classes.txt data/obj.names                                    
%mkdir -p data/obj                                    
#copy image and labels                                    
%cp {dataset.location}/train/*.jpg data/obj/                                
%cp {dataset.location}/valid/*.jpg data/obj/ 
%cp {dataset.location}/train/*.txt data/obj/                                  
%cp {dataset.location}/valid/*.txt data/obj/                                  
                               
                                    
with open('data/obj.data', 'w') as out:                                   
  out.write('classes = 2\n')                                    
  out.write('train = data/train.txt\n')                                   
  out.write('valid = data/valid.txt\n')                                   
  out.write('names = data/obj.names\n')                                   
  out.write('backup = backup/')                                   
                                    
#write train file (just the image list)                                   
import os                                   
                                    
with open('data/train.txt', 'w') as out:                                    
  for img in [f for f in os.listdir(dataset.location + '/train') if f.endswith('jpg')]:                                   
    out.write('data/obj/' + img + '\n')                                   
                                    
#write the valid file (just the image list)                                   
import os                                   
                                    
with open('data/valid.txt', 'w') as out:                                    
  for img in [f for f in os.listdir(dataset.location + '/valid') if f.endswith('jpg')]:                                   
    out.write('data/obj/' + img + '\n') 

Answer 1

确保您的标签(注释)和培训图像名称是相同的。如果标签和图像名称之间存在差异，那么它将导致这个特定的错误。我在处理注释文件时也犯了同样的错误。对我来说，当我不得不从.png转换到.jpg时，就发生了这种情况，这改变了进程中的文件名。