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

我有一个图像张量，我想用python创建它的二进制掩码。 也就是说，将大于零的值替换为255。 有人能适当地建议或指导我吗？ 提前谢谢你

你可以试试，

mask=tf.符号（图像）*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)))

---

如果您有一个布尔掩码，并希望将其应用于rgb图像，则可以使用

tf.where（）

---

这将根据遮罩遮住像素。

tensorflow中是否有其他选项来替换除零以外的值。例如，如果需要替换大于10的值，请说明

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