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

Question

我想使用INCEPTION_V4模型作为特征提取器进行迁移学习 并从以下位置下载了代码和检查点文件 GitHub repository of Tensorflow

然后，我在5个类中添加了自己的分类层。但是在使用tf.train.Saver进行模型还原期间，它显示错误，即找不到我添加的图层的变量值。

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

Answer 1

简单的解决方案是仅构建初始模型（不包含您的图层），然后创建保护程序并将其用于还原，然后再创建您的图层。

还有更复杂的解决方案（您可以将var_list的{​​{1}}参数与要还原的所有变量的列表一起传递，并将其初始化为所有初始变量的列表），但是这应该很简单，我可以使用我的迁移学习模型来做到这一点。