如何将输入输入到张量流预训练模型中的一层？

Question

预训练模型有很多层，我想将我的输入直接输入到一个中间层（并丢弃前一层的结果）。 我只得到了那个模型的.pb文件和ckpt文件，那么如何在没有源代码的情况下修改计算流程呢？

这是我得到的唯一代码文件，但我不知道如何使用它。图是这个文件生成的吗？（和普通的tensorflow文件有很大区别）https://github.com/tensorflow/models/blob/master/research/object_detection/models/ssd_mobilenet_v2_feature_extractor.py

Answer 1

这是你需要做的：

• 加载模型
• 找到层的名称或检索要向其提供值的层的张量（为了便于说明，我们将其命名为“Z”）
• 查找层的名称或检索要从 ('Y') 获取结果的层的张量
• 运行此代码 sn -p :results = sess.run('Y:0', {'Z:0': your_value})