使用 TensorFlow .pb 图作为 Keras 模型

Question

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

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

这可能吗？

Answer 1

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