如何计算张量流中图像中每个通道的均值

Question

在张量流中计算图像中每个通道的平均值的正确方法是什么？

非常感谢任何帮助!!

Answer 1

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

  

轴：要减少的尺寸。如果为None（默认值），则全部减少   尺寸。

Answer 2

一种简单的方法是使用tf.nn.moments

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

其中axes = [0,1,2]

Answer 3

对于形状为[W，H，3]的图像张量，对我有用的是@Rubens_Z答案的导数

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

这将返回[1x3]平均值张量。