способ создания 3-канальной маски из 2d - PullRequest
0 голосов
/ 01 июля 2018

У меня есть числовой массив scale_down, представляющий изображение RGB (форма (500,500,3)), и маска 2d пустого массива (форма (500,500)), и мне нужно заполнить все пиксели scaled_down, где out больше заданного значения, серым цветом , В настоящее время я делаю это следующим образом.

for i in range(out.shape[0]):
  for j in range(out.shape[1]):
    if out[i][j] > 10:
       scale_down[i][j][:] = 100

Но я ожидаю, что у numpy есть еще один способ сделать это без необходимости в циклах. Любая помощь с этим?

1 Ответ

0 голосов
/ 01 июля 2018

Я повторяю здесь ответ, предоставленный @Warren Weckesser.

scale_down[out > 10] = 100
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...