подушка и Numpy, получая значение пикселей

Question

По какой-то причине

im_data = np.array(im.getdata()).reshape(im.size[0], im.size[1], 3) 
p1 = im.getpixel((i, j)) 
p2 = im_data[i, j] 

p1 и p2 являются тем же значением RGB большой частью времени, кроме случаев, когда они не. Любая идея почему?

Answer 1

Причина в том, что numpy работает на основе столбцов, а PIL работает на основе строк при преобразовании одномерных массивов в матрицы или наоборот. Это означает, что функция getdata помещает пиксель из позиции (1,0) во второе место в массиве, а numpy помещает второй пиксель в массив в положение (0,1). Таким образом, вам нужно изменить порядок размеров в режиме изменения и изменить обратное i и j в сравнении. Приведенный ниже код показывает, что тогда нет различий.

Причина, по которой большинство ваших пикселей равны, является совпадением и зависит от вашего изображения. Я попробовал это с фотографией, и почти все пиксели были разными, за исключением пикселей по диагонали.

import numpy as np 
from PIL import Image 


def reshape_img(img: Image): 
    img_data = np.array(img.getdata()).reshape(img.size[1], img.size[0], 3) 
    difference_found = False 
    for i in range(img.size[0]): 
     for j in range(img.size[1]): 
      get_pixel = img.getpixel((i, j)) 
      data = img_data[j, i] 

      if any(get_pixel != data): 
       difference_found = True 
       msg = 'Difference in pixel {pixel}: img.getpixel={getpixel}, ' \ 
         'img_data={data}'.format(pixel=(i, j), getpixel=get_pixel, data=data) 
       print(msg) 
    if not difference_found: 
     msg = 'The two images are identical' 
     print(msg) 


if __name__ == '__main__': 
    ams = Image.open('amsterdam_small.jpg') 
    reshape_img(ams)