[Чтение файлов npy, преобразование и сохранение в формате png] (пакетное сохранение и изображение в градациях серого)

import matplotlib.pyplot as plt
import numpy as np
from scipy import misc
import os
from PIL import Image
import cv2


file_dir = "/dml/graduate_project/data/x_test.npy"  # npy文件路径
dest_dir = "/dml/graduate_project/data/X_test/"  # 文件存储的路径

def npy_png(file_dir, dest_dir):
    file = file_dir  # .npy文件名
    con_arr = np.load(file)
    count = 0 
    for con in con_arr:
        arr=con
        arr=arr *255
        cv2.imwrite(dest_dir+ "_" + '{}'.format(count) + ".png",arr)  
        count = count + 1

if __name__ == "__main__":
    npy_png(file_dir, dest_dir)

При выполнении этой работы она должна быть очень простой, но я столкнулся со многими проблемами.Во
-первых, мой файл .npy хранит изображение в градациях серого с номером канала 1, поэтому при использовании модуля Image для преобразования

def npy_png(file_dir, dest_dir):
    file = file_dir + 'bird.npy'  # .npy文件名
    con_arr = np.load(file)
    count = 0  # 序号，用作设置文件名
    for con in con_arr:
        arr=con
        im = Image.fromarray(arr)
        im = im.convert('L')  # 转为灰度图
        im.save(dest_dir + "_" + '{:06d}'.format(count) + ".png")  
        count = count + 1

if __name__ == "__main__":
    npy_png(file_dir, dest_dir)

Помните навсегда! ! ! !
Большая яма! ! ! ! ! ! ! ! ! !
save_image() не может сохранить реальное изображение в градациях серого, подробности см. в этой статье.

Во-вторых, изображение, сохраненное с помощью cv2.imwrite, в начале полностью черное, поэтому вам нужно добавить предложение arr=arr *255, подробности см. здесь .