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

Question

我有一个张量的图像，我想在 python 中创建它的二进制掩码。那就是用 255 替换大于零的值。有人可以适当地建议或指导我吗？先感谢您

标签： pythontensorflow

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)))