else:
img = Image.open(path2)
# 判断是否为彩色图片
if np.shape(img_array.shape)[0] != 3:
print('p', path2)
os.remove(path2)
# 分别打印出图片对应种类名,以及每类的图片张数
for item1 in os.listdir(path):
path1 = path + item1 + '/'
print(item1)
t=0
for item2 in os.listdir(path1):
t += 1
print(t)
# 图片所在文件夹
path = './fruit_flower/'
# 生成测试集与训练集文件夹
newpath = './fruit_flower/train/'
testpath = './fruit_flower/test/'
# 判断是否存在文件夹
if not os.path.exists(newpath):
os.mkdir(newpath)
if not os.path.exists(testpath):
os.mkdir(testpath)
# 逐个文件夹进行遍历
for items1 in os.listdir(path):
path1 = path + items1 + '/'
newpath1 = newpath + str(i1) + '/'
testpath1 = testpath + str(i1) + '/'
i = 0
# 判断是否存在文件夹
if not os.path.exists(newpath1):
os.mkdir(newpath1)
if os.listdir(newpath1):
shutil.rmtree(newpath1)
os.mkdir(newpath1)
if not os.path.exists(testpath1):
os.mkdir(testpath1)
if os.listdir(testpath1):
shutil.rmtree(testpath1)
os.mkdir(testpath1)
# 逐个图片遍历
for items2 in os.listdir(path1):
i += 1