报错信息:
If error occurs - run training with flag: -dont_show
Error in load_data_detection() - OpenCV
排错思路1:
检查自己的label和JPEGimages文件夹里面的数据集是不是一一对应的。比如我的数据集中图像为500张bmp文件,标签labels有500个txt文件。如果多一个少一个都不行的。
排错思路2:
划分数据集的问题
附上一份划分数据集的代码。这份代码用于划分训练集(train)和验证集(val),划分训练集和验证集的比例。如果运行这份代码报错,可以直接anaconda使用本地的python环境。
import os
import random
def convert(image_dir,train_ratio=1):
file_list=[]
for file in os.listdir(image_dir):
image_file=os.path.join(image_dir,file)
file_list.append(image_file)
random.shuffle(file_list)
train_count=int(len(file_list)*train_ratio)
train_list=file_list[:train_count]
val_list=file_list[train_count:]
with open('train.txt','w') as f:
f.write('\n'.join(train_list))
with open('val.txt','w') as f:
f.write('\n'.join(val_list))
image_dir=r'D:\1\dataset\JPEGImages'
convert(image_dir)