.npy文件转为指定像素的灰度图/RGB图png

前言

本文以转化《quick draw》数据集为例，介绍如何将.npy转为灰度图（或RGB图）。因为本人要使用easydl训练数据，所以必须把该数据集由28 * 28像素转化为30 * 30像素及以上。

python代码

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

file_dir = "E:/inputplace/"  # npy文件路径
dest_dir = "E:/outputplace/"  # 图片文件存储的路径


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<