获取 Mobilenet v1 TensorFlow 的全连接层

Question

伙计们，我使用的是 tensorflow Mobilenet v1，我重新训练了最后一层以输出 4 个类，并且它可以工作。现在，我正在尝试使用最后一个全连接层（或最后一个特征层而不是 softmax 层）将其馈送到 LSTM 网络中。我只是不知道 mobilenet v1 中我需要的最后一层是哪一层。这是我正在使用的模型：mobilenet_v1 这是我用来查看所有层的代码，但我不知道该选择哪一个：

import tensorflow as tf
graph = tf.Graph()
graph_def = tf.GraphDef()
model_file = '/home/root/model/output_graph.pb'
with open(model_file, "rb") as f:
    graph_def.ParseFromString(f.read())
with graph.as_default():
    tf.import_graph_def(graph_def)
print(graph.get_operations())

Answer 1

似乎我需要使用软件之前的最后一个瓶颈层。该层返回一个 1x1001 矩阵