在tensorflow python中替换值或创建张量掩码

Question

我有一个图像张量，我想在python中创建其二进制掩码。 也就是说，将大于零的值替换为255。 有人可以建议或指导我吗？ 预先谢谢你

Answer 1

您可以尝试，        mask = tf.sign(images)*255（如果图像的值> = 0）

要替换大于阈值的值，可以执行以下操作：

thres = 10
cond = tf.greater(images, tf.ones(tf.shape(images))*thres)
mask = tf.where(cond, tf.ones(tf.shape(images))*255, tf.zeros(tf.shape(images)))

Answer 2

如果您具有布尔型蒙版并将其应用于rgb图像，则可以使用tf.where()

image = tf.where(image_mask, image_rgb, tf.zeros_like(image_rgb))

这将根据蒙版使像素变黑。