Создание изображения из списка

Question

Я пытаюсь создать изображение из списка 1s и 0s в Python.

from PIL import Image 

img1 = Image.open('Image1.gif') #the image is strictly black and white 
img1_size = img1.size 
img1_k = [] 
for i in range(img1_size[0]): 
    for j in range(img1_size[1]): 
     tmp_pix = img1.getpixel((i,j)) 
     if (tmp_pix>127): 
      img1_k.append(1) 
     else: 
      img1_k.append(0) 


img = Image.new('RGB', (img1_size[1],img1_size[0]), "white") 
cmap = {1: (255,255,255), 
     0: (0,0,0)} 
data = [cmap[i] for i in img1_k] 
img.putdata(data) 
img.show()    
img.save('Image2.png') 

Однако вместо исходного изображения:

код производит повернутым и перевернутое изображение:

Я предполагаю, что формат putdata () отличается от того, как я приобрел пиксели для списка. Как я могу получить изображение правильно?

Answer 1

getpixel, putpixel основан на координатах x-y, в то время как getdata, putdata основан на строковом столбце.

Во-первых, вам нужно сделать строку-столбец img_k на основе:

for j in range(img1_size[1]):  # row (y) 
    for i in range(img1_size[0]): # column (x) 
     tmp_pix = img1.getpixel((i,j)) 
     if tmp_pix > 127: 
      img1_k.append(1) 
     else: 
      img1_k.append(0) 

Во-вторых, вам нужно создать х * размера изображения у:

img = Image.new('RGB', (img1_size[0], img1_size[1]), "white") # <-- 
# OR Image.new('RGB', img1.size, "white") 
cmap = {1: (255,255,255), 
     0: (0,0,0)} 
data = [cmap[i] for i in img1_k] 
img.putdata(data) 
img.show()    
img.save('Image2.png') 

---

Кстати, вместо смешивание getpixel и putdata, с использованием getdata и putdata код может быть проще:

from PIL import Image 

img1 = Image.open('Image1.gif') 
data = [(255, 255, 255) if pixel > 127 else (0, 0, 0) for pixel in img1.getdata()] 
img = Image.new('RGB', img1.size, "white") 
img.putdata(data) 
img.show()    
img.save('Image2.png')