将npy文件转换成图片

程序如下

import os
import matplotlib.pyplot as plt
import numpy as np
import scipy.misc
from skimage.transform import resize
# output_directory = os.path.dirname('F:\SCRGAN\data\ceshi/npdata/train')  # 提取文件的路径
# output_name = os.path.splitext(os.path.basename("name.npy"))[0]  # 提取文件名
# arr = np.load('F:\SCRGAN\data\ceshi/npdata/train/section_00_source_train_normal_0000_0_g_25_mm_2000_mV_none.wav.npy')  # 提取 npy 文件中的数组
# disp_to_img = resize( arr ,output_shape=(128, 431))  # 根据 需要的尺寸进行修改
# plt.imsave(os.path.join(output_directory, "{}_disp.png".format(output_name)), disp_to_img, cmap='plasma')  # 定义命名规则，保存图片

path='E:\****\logmel/target_test'

path_list = os.listdir(path)
for filename in path_list:
    f = open(os.path.join(path, filename), 'rb')
    arr = np.load(f)
    # output_directory = os.path.dirname('F:\SCRGAN\data\ceshi/npdata/train')  # 提取文件的路径
    # output_name = os.path.splitext(os.path.basename("name.npy"))[0]  # 提取文件名
    disp_to_img = resize(arr, output_shape=(128, 431))
    plt.imsave(os.path.join('F:\SCRGAN\data/target_testpng', "{}_disp.png".format(filename)), disp_to_img, cmap='plasma')

    print(f)