search

openai / guided-diffusion

MIT License
6.11k stars 813 forks source link

npz文件 #116

Open zouyunpeng666 opened 1 year ago

zouyunpeng666 commented 1 year ago

采样生成的图像保存为npz文件，我想知道该怎么将npz文件还原成图片呢

zouyunpeng666 commented 1 year ago

import numpy as np

from PIL import Image data= np.load('samples_10x64x64x3.npz') k = 0 for i in data['arr_0']: img = Image.fromarray(i) img.save(str(k)+".jpg") k = k+1

wang1528186571 commented 3 months ago

import numpy as np

from PIL import Image data= np.load('samples_10x64x64x3.npz') k = 0 for i in data['arr_0']: img = Image.fromarray(i) img.save(str(k)+".jpg") k = k+1

我遇到了这个问题是为什么？ Traceback (most recent call last): File "/home/wang/Desktop/wjl-project/improved-diffusion-main/iamge.py", line 7, in for i in data['arr_0']: File "/home/wang/Anaconda3/envs/diffusion/lib/python3.8/site-packages/numpy/lib/npyio.py", line 260, in getitem raise KeyError("%s is not a file in the archive" % key) KeyError: 'arr_0 is not a file in the archive'