我有10倍,并在各个折叠2个文件夹face
,并且background
,我想复制列车内不同的褶皱,验证文件夹,以便火车或验证将具有face
,background
从不同的褶皱。
我尝试了以下代码,但是由于Google Colab使用python 3.6.9,所以我不能忽略文件夹存在的错误(如python 3.9),因此出现以下错误:
try:
mkdir(name, mode)
except OSError:
# Cannot rely on checking for EEXIST, since the operating system
FileExistsError: [Errno 17] File exists: 'train/'
#--------------
# Split Dataset
#==============
# we Already have the test set
# !mkdir train
# !mkdir valid
import os, shutil
fileList = os.listdir("NewDataset")
fileList.sort()
for i in [x for x in range(10) if ((x != 9) and (x != 1))]:
# print(fileList[i])
# Train
if(i < 7):
subPathList = glob.glob('NewDataset/'+fileList[i]+'/**/', recursive=False)
for subPath in subPathList:
shutil.copytree(subPath, 'train/')
# Validate
else:
subPathList = glob.glob('NewDataset/'+fileList[i]+'/**/', recursive=False)
for subPath in subPathList:
shutil.copytree(subPath, 'valid/')
解决方案是递归地逐个复制文件:
import os, shutil
for i in [x for x in range(10) if ((x != 9) and (x != 1))]:
# print(fileList[i])
# Train
if(i > 2):
subPathList = glob.glob('/content/NewDataset/'+fileList[i]+'/**/', recursive=False)
for subPath in subPathList:
for im in os.listdir(subPath):
imFullPath = os.path.join(subPath, im)
targetPath = os.path.join('/content/train',subPath.split('/')[-2]+ '/')
# print('Train: ', targetPath)
shutil.copy(imFullPath, targetPath)
# Validate
else:
subPathList = glob.glob('/content/NewDataset/'+fileList[i]+'/**/', recursive=False)
for subPath in subPathList:
for im in os.listdir(subPath):
imFullPath = os.path.join(subPath, im)
targetPath = os.path.join('/content/valid',subPath.split('/')[-2] + '/')
# print('Validate: ', targetPath)
shutil.copy(imFullPath, targetPath)
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句