【问题标题】:How do I get the 'checkpoint' of a retrained inception-v3 model?如何获得重新训练的 inception-v3 模型的“检查点”?
【发布时间】:2016-11-15 10:00:18
【问题描述】:

我正在尝试在 tensorflow-serving 中使用经过重新训练的 inception-v3 模型。但似乎我必须提供一个“检查点”。我想知道如何获得这些“检查点”? retrain.py 给我返回了 retrained_graph.pb。我按照本教程进行操作 (https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/#0)

谢谢!

【问题讨论】:

  • 所有谷歌预训练模型的检查点都可以通过github.com/tensorflow/models/tree/master/slim获得。
  • 您找到解决方案了吗? tensorflow 文档对此并不清楚。
  • 您找到解决方案了吗? @cookiedookie

标签: tensorflow tensorflow-serving


【解决方案1】:

您可能想了解 Tensorflow 1.0 的 API 更改

https://www.tensorflow.org/install/migration

使其能够创建检查点文件而不是 .pb 文件。 另请参阅:

https://www.tensorflow.org/tutorials/image_retraining

【讨论】:

    【解决方案2】:

    以下是如何下载最新版预训练 Inception V3 模型的检查点的示例:

    $ CHECKPOINT_DIR=/tmp/checkpoints
    $ mkdir ${CHECKPOINT_DIR}
    $ wget http://download.tensorflow.org/models/inception_v3_2016_08_28.tar.gz
    $ tar -xvf inception_v3_2016_08_28.tar.gz
    $ mv inception_v3.ckpt ${CHECKPOINT_DIR}
    $ rm inception_v3_2016_08_28.tar.gz
    

    【讨论】:

      猜你喜欢
      • 2017-03-28
      • 2017-03-09
      • 1970-01-01
      • 2023-03-25
      • 2018-01-29
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多