将所有0值图像像素转换为tensorflow中的255

时间:2018-06-24 12:33:23

标签: python opencv tensorflow deep-learning opencv3.0

我是tensorflow的新手,

我需要在3d图像张量上将所有0值像素转换为255

我可以这样在简历中做

img[np.where((img == [0, 0, 0]).all(axis=2))] = [255, 255, 255]

但是,我需要帮助以了解如何在tensorflow中做到这一点。因为我需要在张量流图中做到这一点

1 个答案:

答案 0 :(得分:0)

您可以这样做:

img = tf.cast(tf.equal(img, tf.zeros_like(img)), tf.int32)*255 + img