【问题标题】:How to convert a Tensorflow session to a keras model object?如何将 Tensorflow 会话转换为 keras 模型对象?
【发布时间】:2019-06-02 23:36:23
【问题描述】:

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

【问题讨论】:

    标签: tensorflow session keras deep-learning


    【解决方案1】:

    我认为使用 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

    【讨论】:

      猜你喜欢
      • 2021-04-08
      • 1970-01-01
      • 1970-01-01
      • 2018-10-08
      • 2019-05-14
      • 2021-11-11
      • 1970-01-01
      • 2017-11-11
      • 1970-01-01
      相关资源
      最近更新 更多