如何將tensor大於某個值爲1,小於某個值爲0

方法很簡單,定義一個簡單的函數,imgTensor是傳入的需要轉換的tensorboundary是這個臨界值,即大於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)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章