如何将一个图的输出作为输入传递给张量流中的另一个图

Question

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

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

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

Answer 1

简单的解决方案是只构建初始模型（不包含您的层），然后创建保护程序并将其用于恢复，然后才创建您的层。

还有更复杂的解决方案（您可以将tf.Saver 的var_list 参数与您要恢复的所有变量的列表一起传递，并将其初始化为所有初始变量的列表），但是这个应该直截了当，我用我的迁移学习模型来做到这一点。