1. 程式人生 > >將numpy矩陣儲存為圖片

將numpy矩陣儲存為圖片

from PIL import Image
import numpy as np

x=np.load("universal.npy")  # x.shape=m*n*3
x=(x-np.min(x))*255/np.max(x)
r,c,_=x.shape
im = Image.new("RGB", (r, c))

for i in range(0, r):
    for j in range(0, c):
        im.putpixel((i, j), (int(x[i][j][0]),int(x[i][j][1]),int(x[i][j][2])))    #將rgb轉化為畫素

im.
save('flag.jpg')