加载npz文件生成灰度图和深度图:
import os from glob import glob import cv2 import imageio import matplotlib.pyplot as plt import numpy as np input_path = "kits_tumor/pred" paths = glob(os.path.join(input_path,"*_*.npz")) out_path = "save_path" for path in os.listdir(input_path): every_file = os.path.join(input_path,path) data = np.load(every_file) image = data['ct'] pred = data['mask'] index = path.split(".")[0] if not os.path.exists(out_path): os.makedirs(out_path) save_image_path = os.path.join(out_path,f'{index}.jpg') ######################### # 生成灰度图 imageio.imwrite(save_image_path,pred) ######################### # 另一种生成的灰度图(强度较小) cv2.imwrite(save_image_path,pred) ######################### # 生成深度图 plt.imsave(save_image_path,pred) ######################### # 展示一张深度图 plt.imshow(pred) plt.show() #########################
npy文件直接加载