import numpy as np import scipy.io as sio import os import random import cv2 as cv from shutil import copyfile def mkdir(path):#創建文件夾 path=path.strip() path=path.rstrip("\\") isExists=os.path.exists(path) if not isExists: print (path+'OK') os.makedirs(path) return True def autosplitfile():#按照規則移動文件夾 dir_rank = 'E:\\IQA\\eye\\fold\\' # Change to your data folder pristine = 'origin' # The folder of reference images to generate different distortions savefole = 'eye' eyefolder = os.listdir(dir_rank + pristine) for smallfolder in eyefolder: print(smallfolder) real = os.listdir(dir_rank + pristine + "/" + smallfolder) result = [] for filename in real: fullfilename = dir_rank + pristine + "/" + smallfolder + "/" + filename result.append(fullfilename) testsavepath = dir_rank + savefole + "/" + smallfolder mkdir(testsavepath) for t in range(len(result)): namelist = result[t].split('/') name = namelist[len(namelist) - 1] if "_0" in name: copyfile(result[t], testsavepath + '/' + name)