如果npy中有多部分信息
import numpy as np
import scipy.misc
from PIL import Image
import os
path = 'npy文件路径/'
dir = os.listdir(path)
for a in dir:
stem , suffix =os.path.splitext(a)
imgs_test = np.load(str(path)+str(a),allow_pickle=True)
data = imgs_test.item()
score = data['score']
mask = data['mask']
class1 = data['class']
channel = mask.shape[0]
mask2 = np.transpose(mask,(1,2,0))
image = Image.fromarray(mask2, mode="RGB")
image.save('保存路径/'+stem+'.jpg')
如果npy中只有图片信息
import numpy as np
import scipy.misc
from PIL import Image
import os
from matplotlib import pyplot as plt
path = 'npy文件路径/'
dir = os.listdir(path)
for a in dir:
stem , suffix =os.path.splitext(a)
imgs_test = np.load(str(path)+str(a),allow_pickle=True)
data = imgs_test.item()
cam = data['cam']
img = tensor_to_np(cam)
im = Image.fromarray(img)
im.save('保存路径/'+stem+'.jpg')