如何在tensorflow中反转BLACK和WHITE（或灰度）图像？ （黑色 - 白色变化）

Question

我想更改一些仅由黑白元素组成的图片（MNIST）。

我可以通过python元素中的这些代码轻松地将黑色转换为白色，白色转换为黑色

def inverseImageBW_array(originalImage):
    temp = 1 - originalImage
    temp = -1.* originalImage
    return temp

然而，当我处理tensorflow代码时，我不能这样做。 我试图通过

来做到这一点

tf.multiply(1- originalImage, -1)

买它不起作用。

我知道我们可以通过tf.image函数轻松改变亮度或对比度，但我认为没有支持黑白反转的功能。

请帮帮我。

Answer 1

不确定为什么某些东西不适合你。如果你的图像是批量的2d张量或3d张量，那么broadcasting应该处理所有事情：

import tensorflow as tf

a = tf.constant([[0.5, 0.3], [0.2, 0.8]])
b = 1 - a
with tf.Session() as sess:
    print sess.run(b)