代码之家  ›  专栏  ›  技术社区  ›  Ash

从graph*元文件加载tensorflow模型

  •  0
  • Ash  · 技术社区  · 6 年前

    从其他问题( 例如 this ),我认为加载计算图时,通常需要三个文件:

     ckp.data-00000-of-00001
     ckp.index
     ckp.meta
    

    何处 *meta 文件描述了所述图的结构。我试着用一个预先训练过的模型(作为参考,MobileNetv2在 here ,但提供的文件如下:

    - a frozen inference graph (frozen_inference_graph.pb)
    - a checkpoint (model.ckpt.data-00000-of-00001, model.ckpt.index)
    

    我现在不想使用冻结图,因为我需要能够微调模型。但是,正如您从上面的描述中看到的, *元 文件丢失,我在他们的存储库中找不到生成图形的相关代码(作为tensorflow的新手,我可能找错了地方)。

    在这里加载模型的最佳方法是什么?如果我加载冻结图并提取 *元 从那里归档?

    0 回复  |  直到 6 年前
    推荐文章