方法很簡單,定義一個簡單的函數,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)