如何计算每个通道平均在tensorflow图像中

Question

什么是计算平均为tensorflow图像中每个通道的正确方法？

任何帮助深表感谢！！

Answer 1

只需使用tf.reduce_mean()并指定轴：

轴：尺寸减少。如果没有（默认值），降低了各个方面。

Answer 2

一个简单的方法来做到这一点是使用tf.nn.moments

def mean(x):
    mm,_=tf.nn.moments(x,axes=[0])
    return mm

其中轴= [0,1，2]

Answer 3

对于形状的图像张量[W，H，3]，什么工作对我来说是@Rubens_Z答案的衍生物

def mean(x):
    mm,_=tf.nn.moments(x,axes=[0,1])
    return mm

这将返回一个[1×3]平均值张量。