python 将训练集的影像和标签数据分一部分给验证集

将裁剪后的训练样本分一部分给验证集 比例大概 3 : 1, 4 ：1吧，根据实际情况自己定

inputdata=r'E:\0126bs\train\images'    # 影像数据
inputdata1=r'E:\0126bs\train\labels'   # 标签数据

output=r'E:\0126bs\valid\images'       
output1=r'E:\0126bs\valid\labels'


image_dir = list(map(lambda x: inputdata+'/' + x,
                        os.listdir(inputdata+'/')))


random.shuffle (image_dir)

for i in range(2000):           #分给验证集的数量
    print(image_dir[i])
    shutil.move(image_dir[i], output)


for file in os.listdir(inputdata1):
    for file1 in os.listdir(output):
        if file[:-4] == file1[:-4]:
            print(file)
            shutil.move(inputdata1 + '/' + file, output1 + '/' + file)