我使用内置 TensorFlow 工具来微调 InceptionV3 模型的最后一层,以使用 this tutorial 对自定义数据集上的项目进行分类。 。这会生成一堆瓶颈和 TensorFlow 图(*.pb 文件)。
我想将 *.pb TF 图导入 Keras,就像使用包含模型权重的 *.hdf5 文件一样。原因是我想在使用此模型时利用一些用 Keras 编写的工具。
这可能吗?
最佳答案
截至 2017 年 7 月,Keras 只能导入 Keras 模型,而不能导入原始 tf 图。这是因为 Keras 有一些元数据,这些元数据本身并未作为 tf 图的一部分保存,并且无法轻松重建。
关于machine-learning - 使用 TensorFlow .pb 图作为 Keras 模型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44896719/