我是tensorflow的新手,
我需要在3d图像张量上将所有0值像素转换为255
我可以这样在简历中做
img[np.where((img == [0, 0, 0]).all(axis=2))] = [255, 255, 255]
但是,我需要帮助以了解如何在tensorflow中做到这一点。因为我需要在张量流图中做到这一点
答案 0 :(得分:0)
您可以这样做:
img = tf.cast(tf.equal(img, tf.zeros_like(img)), tf.int32)*255 + img