在 TFslim 中加载保存的检查点时出错

Question

使用 v0.12.1 版本的 tensorflow，我正在尝试使用 http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz.

提供的检查点微调预训练的 vgg16 模型

我收到以下错误：

W tensorflow/core/framework/op_kernel.cc:975] Not found: Tensor name "Variable" not found in checkpoint files /home/code/tensorflow/vgg-tensorflow/vgg_16.ckpt
     [[Node: save/RestoreV2 = RestoreV2[dtypes=[DT_INT32], _device="/job:localhost/replica:0/task:0/cpu:0"](_recv_save/Const_0, save/RestoreV2/tensor_names, save/RestoreV2/shape_and_slices)]]
W tensorflow/core/framework/op_kernel.cc:975] Not found: Tensor name "Variable" not found in checkpoint files /home/code/tensorflow/vgg-tensorflow/vgg_16.ckpt
     [[Node: save/RestoreV2 = RestoreV2[dtypes=[DT_INT32], _device="/job:localhost/replica:0/task:0/cpu:0"](_recv_save/Const_0, save/RestoreV2/tensor_names, save/RestoreV2/shape_and_slices)]]

需要帮助。几个月前我发布了一个类似的问题 - 我已经过去了但找不到好的解决方案。我使用

中给出的模型定义

tensorflow.contrib.slim.nets

我知道有两个版本的 ckpt 文件..v1 和 v2。这可能是个问题吗？请问如何解决？

Answer 1

模型定义生成的变量与 ckpt 文件中存储的变量不匹配。具体来说，这是第一个变量。通过这样做解决了它：

variables_to_restore = slim.get_variables_to_restore(exclude=['vgg_16/fc6','vgg_16/fc7','vgg_16/fc8'])
print [v.name for v in variables_to_restore]
restorer = tf.train.Saver(variables_to_restore[1:]) # remove first entry  !

在 TFslim 中加载保存的检查点时出错

Error loading saved checkpoint in TFSlim

tensorflow

tf-slim