python加载npz或npy文件 转换为深度图和灰度图

加载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文件直接加载