tensorflow - 给定连体模型图像的输出向量

Question

此页面 ( https://keras.io/examples/mnist_siamese/ ) 重点介绍了我们如何训练 Siamese 模型。该模型将在给定两个图像输入的情况下输出分数。我想要做的是，在推理过程中，给定一个图像，我希望它返回一个代表图像的 128 维向量，我应该如何实现呢？

Answer 1

如果您运行 model.summary()，您将看到所有模型层的摘要。在您的情况下，“模型”似乎是感兴趣的层。然后您可以使用get_layer()方法选择包含 128D 输出的层。最后，您可以提取如下输出。

model.get_layer('model').output