如何获得重新训练的inception-v3模型的“检查点”？

Question

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

谢谢!

Answer 1

您可能希望研究一下TensorFlow1.0的API更改

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

要使其正常工作，请创建检查点文件而不是.pb文件。另请参阅：

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

Answer 2

以下是如何下载最新版本的预先训练的入侵者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