Bagaimana cara mengonversi array numpy ke (dan menampilkan) gambar?

Karena itu saya telah membuat array:

import numpy as np
data = np.zeros( (512,512,3), dtype=np.uint8)
data[256,256] = [255,0,0]

Yang ingin saya lakukan adalah menampilkan satu titik merah di tengah gambar 512x512. (Setidaknya untuk memulai dengan ... Saya pikir saya bisa mencari tahu sisanya dari sana)

Anda dapat menggunakan PIL untuk membuat (dan menampilkan) gambar:

from PIL import Image
import numpy as np

w, h = 512, 512
data = np.zeros((h, w, 3), dtype=np.uint8)
data[0:256, 0:256] = [255, 0, 0] # red patch in upper left
img = Image.fromarray(data, 'RGB')
img.save('my.png')
img.show()

Berikut ini harus bekerja:

from matplotlib import pyplot as plt
plt.imshow(data, interpolation='nearest')
plt.show()

Jika Anda menggunakan notebook / lab Jupyter, gunakan perintah inline ini sebelum mengimpor matplotlib:

%matplotlib inline 

Jalur terpendek adalah untuk digunakan scipy, seperti ini:

from scipy.misc import toimage
toimage(data).show()

Ini membutuhkan PIL atau Bantal untuk dipasang juga.

Pendekatan serupa juga membutuhkan PIL atau Bantal tetapi yang dapat meminta pengunjung yang berbeda adalah:

from scipy.misc import imshow
imshow(data)

Menggunakan pygame , Anda dapat membuka jendela, mendapatkan permukaan sebagai array piksel, dan memanipulasi seperti yang Anda inginkan dari sana. Anda harus menyalin array numpy Anda ke array permukaan, yang akan jauh lebih lambat daripada melakukan operasi grafis yang sebenarnya pada permukaan pygame sendiri.

Bagaimana menampilkan gambar yang disimpan dalam array numpy dengan contoh (berfungsi di notebook Jupyter)

Saya tahu ada jawaban yang lebih sederhana tetapi yang ini akan memberi Anda pemahaman tentang bagaimana gambar benar-benar tenggelam dari array numpy.

Muat contoh

from sklearn.datasets import load_digits
digits = load_digits()
digits.images.shape   #this will give you (1797, 8, 8). 1797 images, each 8 x 8 in size

Tampilan array dari satu gambar

digits.images[0]
array([[ 0.,  0.,  5., 13.,  9.,  1.,  0.,  0.],
       [ 0.,  0., 13., 15., 10., 15.,  5.,  0.],
       [ 0.,  3., 15.,  2.,  0., 11.,  8.,  0.],
       [ 0.,  4., 12.,  0.,  0.,  8.,  8.,  0.],
       [ 0.,  5.,  8.,  0.,  0.,  9.,  8.,  0.],
       [ 0.,  4., 11.,  0.,  1., 12.,  7.,  0.],
       [ 0.,  2., 14.,  5., 10., 12.,  0.,  0.],
       [ 0.,  0.,  6., 13., 10.,  0.,  0.,  0.]])

Buat kosong 10 x 10 subplot untuk memvisualisasikan 100 gambar

import matplotlib.pyplot as plt
fig, axes = plt.subplots(10,10, figsize=(8,8))

Merencanakan 100 gambar

for i,ax in enumerate(axes.flat):
    ax.imshow(digits.images[i])

Hasil:

Apa yang axes.flatharus dilakukan Itu menciptakan enumerator numpy sehingga Anda dapat beralih pada poros untuk menggambar objek pada mereka. Contoh:

import numpy as np
x = np.arange(6).reshape(2,3)
x.flat
for item in (x.flat):
    print (item, end=' ')

Menggunakan bantal dariarray, misalnya:

from PIL import Image
from numpy import *

im = array(Image.open('image.jpg'))
Image.fromarray(im).show()

The Python Imaging Library dapat menampilkan gambar menggunakan Numpy array. Lihatlah halaman ini untuk kode sampel:

• Konversi Antara Numerik Array dan Objek Gambar PIL

EDIT: Seperti catatan di bagian bawah halaman mengatakan, Anda harus memeriksa catatan rilis terbaru yang membuatnya lebih sederhana:

http://effbot.org/zone/pil-changes-116.htm

Tambahan untuk melakukannya dengan matplotlib. Saya merasa berguna melakukan tugas visi komputer. Katakanlah Anda mendapat data dengan dtype = int32

from matplotlib import pyplot as plot
import numpy as np

fig = plot.figure()
ax = fig.add_subplot(1, 1, 1)
# make sure your data is in H W C, otherwise you can change it by
# data = data.transpose((_, _, _))
data = np.zeros((512,512,3), dtype=np.int32)
data[256,256] = [255,0,0]
ax.imshow(data.astype(np.uint8))