报出这个错误,很是费解。。求大神解救
import tensorflow as tf
import os
import numpy as np
from PIL import Image
import pickle
img_path='./flower_photos'
img_size=224
def read_image(img_path):
imgs=[]
labels=[]
num=0
for i in os.listdir(img_path):
if os.path.isdir(os.path.join(img_path,i)):
# print(i)
for j in os.listdir(os.path.join(img_path,i)):
path=os.path.join(img_path,i)+'/'+j
# print(path)
img=Image.open(path)
img=img.resize((img_size,img_size))
img_ndarray=np.asarray(img,dtype='float32')
imgs.append(np.ndarray.flatten(img_ndarray))
num+=1
if i=='daisy':
labels.append(0)
elif i=='dandelion':
labels.append(1)
elif i=='roses':
labels.append(2)
elif i=='sunflowers':
labels.append(3)
elif i=='tulips':
labels.append(4)
# print(imgs)
# print(labels)
# print('一共%d个图片',num)
return imgs,labels,num
def generate_imgpickle(img_path,outfile):
write_file=open(outfile,'wb')
imgs,labels,num=read_image(img_path)
print(imgs,labels,num)
pickle.dump(imgs[0:num],labels[0:num],write_file,-1)
write_file.close()
def load_imgpickle(outfile):
reader=open(outfile)
data=pickle.loads(reader)
print(reader)
reader.close()
if name=='__main__':
generate_imgpickle(img_path,'trainimg.pickle')
load_imgpickle('trainimg.pickle')