我有一个预训练的CNN模型，格式为.pb文件。我可以加载模型并从最后一层提取所有图像的最终向量。现在我想从特定层而不是最后一层提取我的图像的向量。我使用import_graph_def函数来加载模型，但我不知道各层的名称，因为.pb文件很大，我无法打开它。

我如何只运行模型的一部分而不是整体，以便获取到我想要的那一层的向量？

回答：

除了Peter Hawkins在评论中提到的使用tf.Graph.get_operations()方法之外，还可以使用tensorboard来查找你想提取的层的名称。

然后你可以直接使用

graph.get_tensor_by_name("import/layer_name")

来提取你想要的特征。