yolov5数据集的格式转化为yolov7、yolov8数据集的格式
·
该代码是将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')
更多推荐
所有评论(0)