tensorflow - 如何使用 Keras 获取 Inception v3 模型的 pool3 特征？

Question

使用 Tensorflow，我得到一个 2048 维向量作为 pool3 层的输出。但是，使用 Keras 的 include_top=False 会给出一个 8,8,2048 维向量。我如何获得与使用 Tensorflow 的 pool3 输出层相同的向量？

Answer 1

让我们看看 TensorBoard 中的 pool_3 层。

看起来Keras返回的层实际上是mixed_10层输出。

为了获得 pool_3 的 2048-D 特征向量，Inception v3 添加了一个平均池化层。由于它使用了一个 8x8 滤波器，这是对前两个轴的简单平均操作，因此我们可以使用 NumPy 获得这个向量，如下所示:

pooled_vector = numpy.mean(unpooled_vector, axis=(0,1))

其中 pooled_vector 是 2048 维向量，unpooled_vector 是 8x8x2048 向量。