冻结模型Tensorflow - names_to_saveables错误

Question

在tensorflow/examples/image_retraining/retrain.py文件中，我提供了

  

CHECKPOINT_NAME =＆＃34; / TMP / _retrain_checkpoint / vinaychkpt＆＃34;

模型使用retrain.py构建，检查点目录的内容为：

  

$ ls -l / tmp / _retrain_checkpoint   总计22576

     

-rw-r - r-- 1 Vinay wheel 127 Apr 26 22:07 checkpoint

     

-rw-r - r-- 1 Vinay wheel 9062680 Apr 26 22:07 vinaychkpt.data-00000-of-00001

     

-rw-r - r-- 1 Vinay wheel 11092 Apr 26 22:07 vinaychkpt.index

     

-rw-r - r-- 1 Vinay wheel 2475642 Apr 26 22:07 vinaychkpt.meta

按照link的说明，我尝试使用

冻结模型

bazel-bin/tensorflow/python/tools/freeze_graph --input_graph=/tmp/output_graph.pb --output_graph=/tmp/frozen_graph.pb --input_binary=true --input_checkpoint=/tmp/_retrain_checkpoint/vinaychkpt --output_node_names=softmax

然而，我收到错误：

  

TypeError：names_to_saveables必须是字符串映射到Tensors / Variables的字符串名称。不是变量：Tensor（＆＃34; final_retrain_ops / biases / final_biases：0＆＃34;，shape =（6，），dtype = float32）

我将检查点目录中列出的其他文件名传递给--input_checkpoint参数，但它们似乎都没有工作。

我错过了什么，请帮忙。