ImportError: cannot import name ‘py_checkpoint_reader‘ from ‘tensorflow.pyth

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了ImportError: cannot import name ‘py_checkpoint_reader‘ from ‘tensorflow.pyth相关的知识，希望对你有一定的参考价值。

今天在使用tensor flow里的freeze_graph.py来将模型与权重文件合并时出现了“ImportError: cannot import name 'py_checkpoint_reader' from 'tensorflow.pyth”的错误。

这个错误是因为你下载的tensor flow是1.11.0版本的，而你的tensorflow是1.14.0以上的，所以会报这个错误。

我们可以在git时指定版本，指定为1.15，1.14与1.15无差异写法：

git clone -b r1.15 https://github.com/tensorflow/tensorflow.git

或者你可以直接替换：

将

from tensorflow.python.training import py_checkpoint_reader

替换为

import tensorflow as tf

然后将用到py_checkpoint_reader的替换为tf.train

如：

#reader = py_checkpoint_reader.NewCheckpointReader(input_checkpoint)
reader = tf.train.NewCheckpointReader(input_checkpoint)

以上是关于ImportError: cannot import name ‘py_checkpoint_reader‘ from ‘tensorflow.pyth的主要内容，如果未能解决你的问题，请参考以下文章