Установите ВСЕ пиксели на черный, если значения в позиции 0

Question

У меня есть изображения с такой же формой, как:

 img = cv2.imread(file, 0)  # values are 0 - 255 
 mask = cv2.imread(file2, 0) # values are only 0's and 255's

Из данных изображений я хочу проверить, если маска [x, y] = 0, а затем установить img [x, y ] = 0.

Я могу сделать это, выполнив al oop. Но есть ли способ сделать это с помощью numPyi sh способа?

Answer 1

Вам просто нужно создать маску (не такую, как ваша существующая переменная) и применить ее к массиву img для целевых индексов, в которые вы хотите поместить 0. Тогда это так же просто, как:

mask2 = (mask == 0)
img[mask2] = 0

Или

img[mask.astype(bool)] = 0