python将npy文件转换成图片

如果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) #读入.npy文件
    #print(imgs_test)
    data = imgs_test.item()
    #print(data)
    #读取npy文件中各个不同部分的信息，根据自己的npy文件重写
    score = data['score']
    mask = data['mask']
    class1 = data['class']

    channel = mask.shape[0]
    #转换channel，width，height顺序
    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) #读入.npy文件
    data = imgs_test.item()
    #print(data)
    cam = data['cam']
    img = tensor_to_np(cam)
    im = Image.fromarray(img)

    im.save('保存路径/'+stem+'.jpg')