【问题标题】:Pre-trained checkpoints (.chkpt) Vs GraphDef (.pb)预训练的检查点 (.chkpt) 与 GraphDef (.pb)
【发布时间】:2016-09-25 18:09:44
【问题描述】:

我正在尝试通过使用预训练网络来提取新数据集的特征,例如谷歌在张量流中发布的分类图像图形定义.pb (inception-2015-12-05.tgz)。我成功了,因为transfer_learning 有教程,它使用分类图像图形定义.pb (inception_v3.pb) 来提取新数据集的裂缝。

但是,在新版本的预训练模型中,tensorflow 提供了检查点文件(例如 resnet_v1_152.ckpt)而不是 Graph_def(例如 resnet_v1_152.pb)。我想知道如何使用这些检查点文件来提取 transfer_learning 中的特征。谁能给我一些指示?

【问题讨论】:

    标签: python-2.7 tensorflow pre-trained-model


    【解决方案1】:

    只需按照官方模型保存/恢复文档here即可。

    【讨论】:

      猜你喜欢
      • 2018-06-17
      • 2018-02-16
      • 2018-10-16
      • 2019-08-07
      • 2021-10-05
      • 1970-01-01
      • 1970-01-01
      • 2019-08-16
      • 2020-05-22
      相关资源
      最近更新 更多