使用TensorFlow .pb图作为Keras模型

时间:2017-07-04 03:48:50

标签: machine-learning tensorflow keras

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

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

这可能吗?

1 个答案:

答案 0 :(得分:2)

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