Làm thế nào để chuyển đổi một mảng numpy sang (và hiển thị) một hình ảnh?

Question

Tôi đã tạo ra một mảng thusly:

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

Những gì tôi muốn điều này làm là hiển thị một dấu chấm đỏ duy nhất trong trung tâm của một hình ảnh 512x512. (Ít nhất để bắt đầu ... Tôi nghĩ mình có thể tìm ra phần còn lại từ đó)

Answer 1

Bạn có thể sử dụng để tạo ra PIL (và hiển thị) một hình ảnh:

from PIL import Image 
import numpy as np 

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

Answer 2

Sử dụng pygame, bạn có thể mở cửa sổ, lấy bề mặt làm mảng pixel và thao tác theo ý muốn từ đó. Tuy nhiên, bạn sẽ cần phải sao chép mảng numpy của bạn vào mảng bề mặt, điều này sẽ chậm hơn nhiều so với các hoạt động đồ họa thực tế trên các bề mặt pygame.

Answer 3

Python Imaging Library có thể hiển thị hình ảnh bằng mảng Numpy. Hãy xem tại trang này cho mẫu mã:

• Convert Between Numerical Arrays and PIL Image Objects

EDIT: Như ghi chú trên dưới cùng của trang đó nói, bạn nên kiểm tra ghi chú phát hành mới nhất mà làm này đơn giản hơn nhiều:

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

Answer 4

con đường ngắn nhất là sử dụng scipy, như thế này:

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

Điều này cũng cần cài đặt PIL hoặc Gối.

Một cách tiếp cận tương tự cũng đòi hỏi PIL hoặc gối nhưng mà có thể gọi một người xem khác nhau là:

from scipy.misc import imshow 
imshow(data) 

Answer 5

Bạn chỉ có nghĩa là điều này?

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