npy2jpg

import os
import glob
import numpy as np
from PIL import Image
out_dir = './npy2jpg'
if not os.path.exists(out_dir):
    os.mkdir(out_dir)
npy_file = glob.glob(r"./*.npy")
npy_file.sort()
for index,item in enumerate(npy_file):
    file = np.load(item)
    image = Image.fromarray(file[:,:,0:3])
    outfile = os.path.join(out_dir,npy_file[index][2:-4]+'.jpg')
    image.save(outfile)