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