Маскировка изображения BGR с использованием ошибки 2D-маски при отображении

Question

Я пытаюсь замаскировать цветное изображение с помощью 2D-маски в python.Я пробовал решение, приведенное в ответе здесь Маскирование изображения BGR с использованием 2D-маски , но оно не отображает желаемый результат.

Я пробовал этот код из предыдущего ответа:

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()

masked_image = image[mask]

, где изображение имеет форму (522, 775, 3), а image_mask имеет форму (522, 775) и 0 или 1 в качестве значений.

masked_image.shape 

выводит это (522, 775, 3, 775, 3) и, следовательно, изображение не отображается.

Ожидаемым результатом будет замаскированное изображение с нулями в местах, гдемаска также равна 0.Вместо этого я получаю странную форму в результирующем masked_image и не могу понять почему.Любая помощь будет высоко ценится!

Answer 1

Что произойдет, если вы умножите свое изображение с маской?

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()

masked_image = image * mask