2017-06-14 109 views

回答

1

只需使用tf.reduce_mean()並指定軸:

軸:尺寸減少。如果無(默認),則縮小所有維度。

+0

我想reduce_mean會發現每個RGB像素的平均沿着第三維的值,即R + G + B/3。那麼如果想要執行紅色通道中所有像素的平均值,同樣在藍色和綠色通道中。謝謝@薩爾瓦多 –

+0

@ I.Ayoub減少意味着與圖像無關。它適用於任何張量,無論它代表什麼。你只需要找到適當的軸,你想要計算平均值。寫一個簡單的張量,您可以手動進行計算並測試您需要選擇哪個軸 –

0

一個簡單的方法來做到這是使用tf.nn.moments

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

其中軸= [0,1,2]