分割数据集为训练集和验证集合

import os
import random
import shutil

#原来的图片路径
src_img_dir = r".\data\target"
#原来的标签路径
src_label_dir = r".\data\label"
#目标图片路径
target_img_dir = r".\mydata\img"
#目标标签路径
target_label_dir = r".\mydata\label"

img_path = os.listdir(src_img_dir)

list = []
for name in img_path:
    list.append(name)
#打乱顺序
random.shuffle(list)

for index, name in enumerate(list):
    # 按照8，2分
    if index <= len(list) * 0.8:
        target_str = "train"
    else:
        target_str = "val"
    src_dir = os.path.join(src_img_dir, name)
    target_dir = os.path.join(target_img_dir, target_str, name)
    label_name = name.replace(".jpg", ".png")
    src_label = os.path.join(src_label_dir, label_name)
    target_label = os.path.join(target_label_dir, target_str, label_name)
    print(src_dir)
    print(target_dir)
    print(src_label)
    print(target_label)
    shutil.copyfile(src_dir, target_dir)
    shutil.copyfile(src_label, target_label)

img 后缀jpg,label后缀png