在tensorflow中将所有0值图像像素转换为255
Convert all 0 value image pixels to 255 in tensorflow
我是tensorflow新手,
我需要将 3d 图像张量上的所有 0 值像素转换为 255
我可以这样写简历
img[np.where((img == [0, 0, 0]).all(axis=2))] = [255, 255, 255]
但是,我需要帮助来了解如何在 tensorflow 中执行此操作。因为我需要在张量流图中这样做
你可以这样做:
img = tf.cast(tf.equal(img, tf.zeros_like(img)), tf.int32)*255 + img
我是tensorflow新手,
我需要将 3d 图像张量上的所有 0 值像素转换为 255
我可以这样写简历
img[np.where((img == [0, 0, 0]).all(axis=2))] = [255, 255, 255]
但是,我需要帮助来了解如何在 tensorflow 中执行此操作。因为我需要在张量流图中这样做
你可以这样做:
img = tf.cast(tf.equal(img, tf.zeros_like(img)), tf.int32)*255 + img