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

Question

Допустим, у меня есть изображение OpenCV в виде массива с нулевыми значениями. Я хочу собрать все (x, y) координаты точек точек, которые имеют больше нуля синего компонента цвета. Наивный путь был бы примерно таким:

n_rows, n_cols, _ = image.shape
points = []
for row in range(n_rows):
    for col in range(n_cols):
        if image[row, col, 0] > 0:
            points.append((row, col))

Есть ли более умный и эффективный способ сделать то же самое, используя возможности numpy или OpenCV?

Answer 1

Это поможет:

np.argwhere(image[:,:,0] > 0)

Для данного image, имеющего три цветовых канала (в этом BGR), 0 в image[:,:,0] обозначает первый канал (синий)

Сам оператор возвращает массив координат, где значения пикселей синего канала больше 0