在tensorflow中如何将一个图的输出作为输入传递给另一个图

Question

我想用INCEPTION_V4模型作为特征提取器进行传输学习，并从Tensorflow的GitHub存储库下载代码和检查点文件。

然后，我在5个类中添加了自己的分类层。但是在使用tf.train.Saver进行模型恢复时，它显示出错误，无法找到我添加的层的变量值。

为了解决这个问题，我创建了两个独立的图，一个用于加载预先训练的模型，另一个用于我的分类层。但是我不能把一个图的输出作为第二个图的输入。你能建议其他方法来转移学习或解决问题吗？

Answer 1

简单的解决方案是只构建初始模型(没有层)，然后创建保护程序并使用它进行还原，然后才创建层。

还有更复杂的解决方案(您可以将tf.Saver的tf.Saver参数与要还原的所有变量列表一起传递，并将其初始化为所有初始变量的列表)，但这一项应该简单明了，我使用传输学习模型来实现。