当我使用TF-slim中的eval_image_classifier.py使用我的数据集评估重新调整的inception-v3模型时,我收到了一个错误:
NotFoundError (see above for traceback): Key InceptionV3/AuxLogits/Conv2d_2a_3x3/weights not found in checkpoint
[[Node: save/RestoreV2_7 = RestoreV2[dtypes=[DT_FLOAT], _device="/job:localhost/replica:0/task:0/device:CPU:0"](_arg_save/Const_0_0, save/RestoreV2_7/tensor_names, save/RestoreV2_7/shape_and_slices)]]
然后我打印了所有张量'在orignal inception-v3检查点和重新调整的检查点中的名称,然后将这些张量与张量inception_v3.arg_scope进行比较,我发现检查点中的一些张量与inception_v3.arg_scope中的张量不同:
"InceptionV3/AuxLogits/Conv2d_2a_3x3" in the inception_v3.arg_scope
"InceptionV3/AuxLogits/Conv2d_2a_5x5" in the checkpoints
之前是否改变了inception_v3的架构?如何解决这个问题呢?使用inception_v3.arg_scope重新调整整个inception-v3模型而不是重新调整?
答案 0 :(得分:0)
将tf.reset_default_graph()
添加到python脚本,它将解决此问题。