方法很简单,定义一个简单的函数,​​imgTensor​​​是传入的需要转换的​​tensor​​​,​​boundary​​​是这个临界值,即大于​​boundary​​​的值设为​​1​​​,小于​​boundary​​​的值设为​​0​​​。本函数使用​​tf.where​​​来实现,那么很容易可以看出我们可以任意设置​​tensor​​​大于某个值(或小于某个值)时的值,只需要将下面代码中的​​tf.ones_like​​​(或:​​tf.zeros_like​​​)改成​​tf.fill()​​​来构造一个相同大小的​​tensor​​即可。

def getBinaryTensor(imgTensor, boundary = 200):
one = tf.ones_like(imgTensor)
zero = tf.zeros_like(imgTensor)
return tf.where(imgTensor > boundary, one, zero)