4

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

谢谢!

4

2 回答 2

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
于 2017-05-02T09:58:25.100 回答
0

您可能想查看 Tensorflow 1.0 的 API 更改

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

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

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

于 2017-02-24T22:49:03.017 回答