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