我正在尝试在 tensorflow-serving 中使用经过重新训练的 inception-v3 模型。但似乎我必须提供一个“检查点”。我想知道如何获得这些“检查点”?retrain.py
返回给我一个retrained_graph.pb
. 我按照本教程(https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/#0)
谢谢!
我正在尝试在 tensorflow-serving 中使用经过重新训练的 inception-v3 模型。但似乎我必须提供一个“检查点”。我想知道如何获得这些“检查点”?retrain.py
返回给我一个retrained_graph.pb
. 我按照本教程(https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/#0)
谢谢!
以下是如何下载最新版预训练 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
您可能想查看 Tensorflow 1.0 的 API 更改
https://www.tensorflow.org/install/migration
使其能够创建检查点文件而不是 .pb 文件。另请参阅: