получение значения в пикселях для изображения, проиндексированного по меткам

Question

У меня есть RGB-изображение Img(256,256,3), которое было сегментировано.Метки для этого изображения находятся в массиве Lbl(256,256) со значениями меток в диапазоне от 0-n, где n - количество кластеров в изображении.Как получить фактические значения RGB для пикселей, назначенных конкретному кластеру?Например, как мне найти все значения пикселей, назначенные кластеру 1?

Я уверен, что в Numpy есть очень питонический способ сделать это.

Answer 1

Вы можете сделать это с помощью np.where() следующим образом:

import numpy as np

# Make sample empty image
a = np.zeros((8,8),dtype=np.uint8)                                                         

# Label a couple of random pixels as "3" to find
a[2,2]=3                                                                                   
a[3,4]=3                                                                                   

# Find them
my3s = np.where(a==3)                                                                             
Out[13]: (array([2, 3]), array([2, 4]))