【发布时间】:2019-06-02 23:36:23
【问题描述】:
假设我有一个预训练模型存储在 Tensorflow 检查点中。我想将其转换为 Keras 模型。我可以将检查点加载到 TF 会话中,但这就是我卡住的地方。
【问题讨论】:
标签: tensorflow session keras deep-learning
假设我有一个预训练模型存储在 Tensorflow 检查点中。我想将其转换为 Keras 模型。我可以将检查点加载到 TF 会话中,但这就是我卡住的地方。
【问题讨论】:
标签: tensorflow session keras deep-learning
我认为使用 TF 检查点创建 Keras 模型是不可能的,但您可以将其权重复制到已经创建的 Keras 模型中。
检查一下。 https://github.com/yuyang-huang/keras-inception-resnet-v2
extract_weights.py 用于将 TF 权重保存到 numpy 数组,而load_weights.py 用于将npy 文件加载到 Keras 模型中。
更多参考,这是我实现它的方法https://github.com/DableUTeeF/keras-efficientnet/tree/master/keras_efficientnet。
【讨论】: