如何在tensorflow中合并两个模型？

Question

我正在使用tensorflow进行人脸识别。 我有一个图表，我用来训练两个不同的模型，每个模型有10个类。 现在我的问题是，我可以借助这两个模型创建一个包含20个类的模型。 请回答这个问题。如果您想了解更多信息，请与我们联系。 提前谢谢。

Answer 1

我不建议你合并两个图表，主要是因为20个类别的参与者与每10个类别的合并参数完全不同，但是如果你还想这样做，这里有一些建议

首先使用meta_graph_def或graph_def

导入图表

图中的第二个提取“瓶颈”张量，你可以用bottleneck = graph.get_tensor_by_name("your_tensor_name")

来做

第三个从磁盘（检查点）读取预训练值，类tf.train.NewCheckpointReader可以帮助

最后一步，在瓶颈张量上建立一个常数张量，从第三步读取值。