Numpy, где используются различные размеры массива

Question

Итак, у меня есть два изображения PIL RGBA. То, что я хочу сделать, это найти все места, где значения RGB являются одинаковыми и альфа 255. выглядит следующим образом:

from PIL import Image 
import numpy as np 
img1 = np.array(Image.open(/path/to/img1).convert('RGBA'), float).reshape(32,32,4) 
img2 = np.array(Image.open(/path/to/img2).convert('RGBA'), float).reshape(32,32,4) 

# Checks to see if RGB of img1 == RGB of img2 in all locations that A=255 
np.where((img1[:,:,:-1] == img2[:,:,:-1]) &\ # RGB check 
     (img1[:,:,3] == 255)) # Alpha check 

Но это приводит к operands could not be broadcast together with shapes (32,32,3) (32,32).
Я не думал, что я пытался передать их вместе, я просто хотел найти детексы, которые, я думаю, в свою очередь транслирует их в этом заявлении. Есть ли другой способ сделать это или способ не транслировать неравные формы?

Answer 1

Используйте .all(axis=-1), чтобы найти места, где все три значения RGB равны:

np.where((img1[..., :-1] == img2[..., :-1]).all(axis=-1) 
     & (img1[..., 3] == 255)) 

---

Как mgilson points out, (img1[..., :-1] == img2[..., :-1]) имеет форму (32, 32, 3). Вызов .all(axis-1) уменьшает последнюю ось к скалярному логическому значению, так что

(img1[..., :-1] == img2[..., :-1]).all(axis=-1) 

имеет форму (32, 32). Это соответствует форме (img1[..., 3] == 255), поэтому эти булевы массивы можно комбинировать с bitwise-and operator, &.