我从一组图像中提取了特征。 Keras功能提供了形状python tf_program2.py
,其中X是图像的数量。
我可以看到有512层。每层是(7,7)。我需要访问每一层中49个元素中的每一个并取其平均值。但是我无法理解如何以上述格式访问元素。
我该怎么做呢?有人能帮我清楚吗?
答案 0 :(得分:0)
您可以使用np.mean
指定轴计算每个图层的平均值,np.mean(axis=(1, 2))
。为了确保这样做你想要的,你可以测试几个层:
import numpy as np
data = np.random.randn(10, 7, 7, 512)
layer_means = np.mean(data, axis=(1, 2))
print(layer_means[0, 0], np.mean(data[0, :, :, 0]))
print(layer_means[2, 6], np.mean(data[2, :, :, 6]))