制作真伪数据集脚本

#将文件夹中的图片归类到对应文件夹的子文件夹中,比如将大train中的train_real和train_fake中名称对应的图片转移到Image中的train的子文件夹对应的名称文件夹中(train Image test文件夹并列),具体的文件夹分布在微信收藏夹里面

import os
import shutil

def read_filename():
    file_path='F:/dealwith/train/train_fake'
    list=os.listdir(file_path)
    return list

def copy_file(file_list):
    for i in file_list:
        dst_path='F:/dealwith/Image/train/'+i[:-4]+'/real/'
        src_path='F:/dealwith/train/train_fake/'+i
        print("拷贝"+i)
        shutil.copy(src_path,dst_path)


def read_filename_train():
    file_path='F:/dealwith/train/train_real'
    list=os.listdir(file_path)
    return list

def copy_file_train(file_list):
    for i in file_list:
        temp=str(int(i[:-4]))
        src_path='F:/dealwith/train/train_real/'+i
        dst_path='F:/dealwith/Image/train/'+temp+'/real/'
        print("copy "+i+' to '+dst_path)
        shutil.copy(src_path,dst_path)



copy_file_train(read_filename_train())