Как выбрать канал с максимальным значением на каждый пиксель в многоканальном изображении?

Question

Я хочу получить канал с максимальным значением для каждого пикселя в многоканальном изображении.Я хочу сделать с NumPy без использования цикла.Есть ли ярлык для извлечения этих значений?

Я могу сделать то же самое, зацикливаясь на каждом значении пикселя, как показано ниже:

label_list = []
for i in range(height):
    for j in range(width):
        label = np.where(img[i,j,:] == np.max(img[i,j,:]))
        if label not in label_list:
            label_list.append(label)

Каждый канал в моем изображении соответствует одному объекту.Я хочу знать, какие объекты присутствуют на изображении.

Answer 1

Argmax может решить цель.Возвращает индексы максимального значения по оси.

np.unique (np.argmax (img, axis = 2))

Может быть, это может помочь кому-то еще.