将GAN生成的图像转移到train文件夹中

将GAN生成的图像转移到train文件夹中

import os
import shutil

sour_path='./samples/new-demo/'
train_path='./data/*****/datasets/train/'
# 两个文件夹结构要相同，例如new-demo/(*)/*.bmp,train/*。
for file in os.listdir(sour_path):
    folder = os.path.join(sour_path, file)
    if os.path.isdir(folder):
        for i in range(100):
            # 将前100张放在train文件夹下
            img = folder + '/' + str(i+1) +'.bmp'
            if os.path.exists(img):
                shutil.copy(img, train_path+'(' + file+')' + '/(' + str(i+4) + ')' + '.bmp')