Tensorflow js подсчет черных пикселей изображения

Question

Я читаю изображения RGB в тензорном потоке. js и мне нужно узнать количество черных пикселей [0,0,0] в этом изображении? Есть ли способ сделать эту операцию?

Answer 1

Тензорное изображение - это трехмерный тензор без прозрачности

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

t = tf.tensor([0, 0, 0, 1, 0, 0, 0, 0, 0, 2, 1, 0], [2, 2, 3])
t.sum(-1).equal(tf.zeros(t.shape.slice(0, -1))).sum().print() // 2

Мы суммируем каждое значение пикселя (сумма по ось -1) и создайте двумерный тензор той же ширины и высоты, что и исходный тензор. Оба тензора сравниваются, и последний tf.sum будет считаться, где первый tf.sum тензор имеет 0 значений.