如果npy中有多部分信息

import numpy as np import scipy.misc from PIL import Image import ospath = '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文件中各個不同部分的信息，根據(jù)自己的npy文件重寫score = data['score']mask = data['mask']class1 = data['class']channel = mask.shape[0]#轉(zhuǎn)換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 pltpath = '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')

總結(jié)

以上是生活随笔為你收集整理的python将npy文件转换成图片的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。