Tensorflow:如何减去平均图像？

Question

我正在寻找有关如何从输入图像中减去平均图像的代码片段。我将图像表示为一个 numpy 数组:

IMG_MEAN = np.array((104.00698793,116.66876762,122.67891434), dtype=np.float32)

我猜减法部分会是这样的

img = tf.subtract(img, mean_img)

但是如何创建与 img 具有相同形状的 mean_image 呢？

Answer 1

假设 img 都是一个矩阵，大小为 (100, 100, 3)rgb 平均值是 [1, 2, 3]

img = tf.ones([100, 100, 3], dtype=tf.float32) #(100, 100, 3)
mean = tf.constant([1, 2, 3], dtype=tf.float32) # (3)
mean = tf.reshape(mean, [1, 1, 3])

img_m = img - mean

with tf.Session() as sess:
    a = sess.run(img_m) # shape of a (100, 100, 3)
    # a[:,:,0] = 0
    # a[:,:,1] = -1
    # a[:,:,1] = -2