Проблема с применением бинарной маски к изображению RGB с помощью numpy ⇐ Python

[Anonymous]

Я пытаюсь применить двоичную маску к изображению RGB с помощью numpy.

Я нашел это https://stackoverflow.com/a/26843467/4628384, но у меня пока нет разрешения на написание комментария.
В любом случае, у меня возникла проблема; Любая помощь очень ценится.

Код: Выделить всё

def masktoRGB(self,image,image_mask):

# create mask with same dimensions as image
mask = np.zeros_like(image)
# copy your image_mask to all dimensions (i.e. colors) of your image
for i in range(3):
mask[:,:,i] = image_mask.copy()

# apply the mask to your image

# tried to swap axes, not a solution
#image = image.swapaxes(0,1)
#this gives the error:
masked_image = image[mask]

print(mask.shape)
print(image.shape)
print(image_mask.shape)

return masked_image

это дает мне:

IndexError: индекс 213 выходит за пределы оси 0 с размером 212

вывод печати:

(188, 212, 3)
(188, 212, 3)
(188, 212)

image и image_mask — это одно и то же изображение, за исключением того, что первое — RGB, а второе — режим L

Подробнее здесь: https://stackoverflow.com/questions/584 ... with-numpy