使用 TensorFlow .pb 图作为 Keras 模型

2024-03-29

我使用内置 TensorFlow 工具来微调 InceptionV3 模型的最后一层,以便对自定义数据集上的项目进行分类本教程 https://www.tensorflow.org/tutorials/image_retraining。这会生成一堆瓶颈和 TensorFlow 图(*.pb 文件)。

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

这可能吗?


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

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

使用 TensorFlow .pb 图作为 Keras 模型 的相关文章

随机推荐