image - 关于使用预训练的 im2txt 模型

Question

我已经遵循了这里的每一步https://edouardfouche.com/Fun-with-Tensorflow-im2txt/但我收到以下错误

NotFoundError(请参阅上面的回溯):在检查点文件/home/asadmahmood72/Image_to_text/models/im2txt/model.ckpt-3000000 中找不到张量名称“lstm/basic_lstm_cell/bias”[[节点:save/RestoreV2_380 = RestoreV2[dtypes=[DT_FLOAT], _device="/job:localhost/replica:0/task:0/cpu:0"](_arg_save/Const_0_0, save/RestoreV2_380/tensor_names, save/RestoreV2_380/shape_and_slices)]]

我的操作系统是 UBUNTU 16.04我的tensorflow版本是1.2.0

Answer 1

这有点晚了，但希望这个答案能够帮助 future 遇到此问题的人。

正如 Edouard 提到的，此错误是由于 Tensorflow API 的更改引起的。如果您想使用更新版本的 Tensorflow，我知道有几种方法可以“更新”您的检查点:

1. 使用官方checkpoint_convert.py Tensorflow 中包含的实用程序，或

2. 使用this solution由 0xDFDFDF 在 GitHub 上编写，用于重命名有问题的变量:

   OLD_CHECKPOINT_FILE = "model.ckpt-1000000"
   NEW_CHECKPOINT_FILE = "model2.ckpt-1000000"
   
   import tensorflow as tf
   vars_to_rename = {
       "lstm/basic_lstm_cell/weights": "lstm/basic_lstm_cell/kernel",
       "lstm/basic_lstm_cell/biases": "lstm/basic_lstm_cell/bias",
   }
   new_checkpoint_vars = {}
   reader = tf.train.NewCheckpointReader(OLD_CHECKPOINT_FILE)
   for old_name in reader.get_variable_to_shape_map():
     if old_name in vars_to_rename:
       new_name = vars_to_rename[old_name]
     else:
       new_name = old_name
     new_checkpoint_vars[new_name] = tf.Variable(reader.get_tensor(old_name))
   
   init = tf.global_variables_initializer()
   saver = tf.train.Saver(new_checkpoint_vars)
   
   with tf.Session() as sess:
     sess.run(init)
     saver.save(sess, NEW_CHECKPOINT_FILE)
   

我使用了选项 #2，之后加载我的检查点就完美了。