博主个人的数据集格式为 voc 数据集,由于之前转v5的时候是用 voc 数据集先转成YOLO格式数据集,这里主要是对YOLOv5数据集进行转换。
images 里的内容:
转换代码:
import glob
#存放图片的地址
train_image_path = r"D:\YOLOv7_Maize\images\train"
valid_image_path = r"D:\YOLOv7_Maize\images\val"
#生成的txt的路径
txt_path = r"D:\YOLOv7_Maize/"
def generate_train_and_val(image_path, txt_file):
with open(txt_file, 'w') as tf:
for jpg_file in glob.glob(image_path + '//' + '*.jpg'):
tf.write(jpg_file + '\n')
generate_train_and_val(train_image_path, txt_path + 'train.txt')
generate_train_and_val(valid_image_path, txt_path + 'valid.txt')