ImportError: cannot import name ‘py

今天在使用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)