使用 TensorFlow .pb 图作为 Keras 模型

Question

我已使用内置 TensorFlow 工具微调 InceptionV3 模型的最后几层，以使用 this tutorial 对自定义数据集上的项目进行分类。这会生成一堆瓶颈和一个 TensorFlow 图（*.pb 文件）。

我想将 *.pb TF 图导入 Keras，就像您使用包含模型权重的 *.hdf5 文件一样。原因是在使用此模型时，我想利用一些用 Keras 编写的工具。

这可能吗？

Answer 1

截至 2017 年 7 月，Keras 只能导入 Keras 模型，不能导入原始 tf 图。这是因为 Keras 有一些元数据，这些元数据本身并没有作为 tf 图的一部分保存，并且不容易重建。