YOLO v7数据集格式 (附代码)

博主个人的数据集格式为 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')