tensorflow - 如何将 Tensorflow 会话转换为 keras 模型对象？

Question

假设我有一个预训练模型存储在 Tensorflow 检查点中。我想将其转换为 Keras 模型。我可以将检查点加载到 TF 会话中，但这就是我卡住的地方。

标签： tensorflowsessionkerasdeep-learning

Answer 1

我认为使用 TF 检查点创建 Keras 模型是不可能的，但是您可以将其权重复制到已经创建的 Keras 模型中。

签出这个。https://github.com/yuyang-huang/keras-inception-resnet-v2

是将extract_weights.pyTF 权重保存到 numpy 数组，而load_weights.py将npy文件加载到 Keras 模型。

如需更多参考，这就是我实现它的方式https://github.com/DableUTeeF/keras-efficientnet/tree/master/keras_efficientnet。