qq_24966011 2018-10-14 01:52 采纳率: 0%
浏览 5133
已结题

TypeError: a bytes-like object is required, not '_io.TextIOWrapper' 想把标签和图片变成pickle

报出这个错误,很是费解。。求大神解救
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')

  • 写回答

1条回答 默认 最新

  • devmiao 2018-10-14 03:40
    关注
    评论

报告相同问题?

悬赏问题

  • ¥15 stata安慰剂检验作图但是真实值不出现在图上
  • ¥15 c程序不知道为什么得不到结果
  • ¥40 复杂的限制性的商函数处理
  • ¥15 程序不包含适用于入口点的静态Main方法
  • ¥15 素材场景中光线烘焙后灯光失效
  • ¥15 请教一下各位,为什么我这个没有实现模拟点击
  • ¥15 执行 virtuoso 命令后,界面没有,cadence 启动不起来
  • ¥50 comfyui下连接animatediff节点生成视频质量非常差的原因
  • ¥20 有关区间dp的问题求解
  • ¥15 多电路系统共用电源的串扰问题