Проблемы, связанные с созданием маски изображения RGB в opencv python

Question

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

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

Я могу предоставить изображение, если оно вообще требуется.

Вот кодовый сегмент

image = cv2.imread("abcd.png")
for k in range(image.shape[0]):
    for l in range(image.shape[1]):
        if(image[k][l]==[255,255,255]):
            mask[k][l]=255
        else:
            mask[k][l]=0

Интересно, в чем проблема в коде?

Answer 1

Итерации по пикселям с for циклами очень медленные - попробуйте привыкнуть к векторизации обработки с помощью Numpy.

import numpy as np
import cv2

# Load image
image = cv2.imread("start.png")

# Mask of white pixels - elements are True where image is White
Wmask =(im[:, :, 0:3] == [255,255,255]).all(2) 

# Save as PNG
cv2.imwrite('result.png', (Wmask*255).astype(np.uint8))

Итак, начиная с этого изображения:

Вы получите эту маску:

Answer 2

В самой ошибке есть подсказка.Вы можете использовать numpy.all(), чтобы проверить, является ли пиксель изображения белым или нет.

Новый код:

import cv2
import numpy as np

image = cv2.imread("image.png")
h, w = image.shape[:2]
mask = np.zeros((h, w))

for k in range(h):
    for l in range(w):
        if np.all(image[k][l]==255): # true if (image[k][l][0]==255 and image[k][l][1]==255 and image[k][l][1]==255)
           mask[k][l]=255