Tensorflowjs计算图像中的黑色像素

Question

我正在tensorflow.js中读取RGB图像，需要找出该图像中黑色像素[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上求和(，并创建一个与初始张量具有相同宽度和高度的2d张量。比较两个张量，最后一个tf.sum将计数，其中第一个tf.sum张量具有0值。