该代码是将yolov5格式的yolo数据集转化为yolov7格式yolo的数据集,yolov5格式比yolov7少了train.txt和val.txt

import glob
 
# 注意这里一定要放着datasets 
#存放图片的地址
train_image_path = r"C:\Users\Administrator\Desktop\yolov7\datasets\images\train"
valid_image_path = r"C:\Users\Administrator\Desktop\yolov7\datasets\images\val"
#生成的txt的路径
txt_path = r"C:\Users\Administrator\Desktop\yolov7\datasets/"
 
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'):
            jpg_file_path = jpg_file[jpg_file.rfind('dataset'):] 
            tf.write(jpg_file_path + '\n')
 
generate_train_and_val(train_image_path, txt_path + 'train.txt')
generate_train_and_val(valid_image_path, txt_path + 'val.txt')
 
 
Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐