如何获得检查点'再培训的初始-v3模型?

时间:2016-11-15 10:00:18

标签: tensorflow tensorflow-serving

我尝试在tensorflow服务中使用重新训练的inception-v3模型。但似乎我必须提供一个检查点'。我想知道如何获得这些检查点'? retrain.py会向我返回retrained_graph.pb。我遵循了本教程(https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/#0

谢谢!

2 个答案:

答案 0 :(得分:0)

您可能需要查看Tensorflow 1.0的API更改

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

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

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

答案 1 :(得分: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