数组中的全局平均池

Question

我正在使用Keras的InceptionV3模型来提取功能。假设我有1000张图片，我得到了最后一层形状(1000, 8, 8, 2048)。数据大小为1000，最后一个卷积层为(8, 8, 2048)。如何实施全球平均合并？我期待形状为(1000, 1, 1, 2048)。

Answer 1

>>> import numpy as np
>>> X.shape
    (1000, 8, 8, 2048)
>>> X_gap = np.apply_over_axes(np.mean, X, [1, 2])
>>> X_gap.shape
    (1000, 1, 1, 2048)