报错InvalidArgumentError (see above for traceback): Restoring from checkpoint failed

最新推荐文章于 2022-11-14 16:42:56 发布

xiaoqiaoliushuiCC

最新推荐文章于 2022-11-14 16:42:56 发布

阅读量1.8k

点赞数

文章标签： tensorflow

本文链接：https://blog.csdn.net/xiaoqiaoliushuiCC/article/details/106932259

版权

报错详情：
使用faster-rcnn训练自己的模型，下载使用别人的代码
这份代码
运行python tools/demo.py报错
如下：

Traceback (most recent call last):
  File "tools/demo.py", line 145, in <module>
    saver.restore(sess, tfmodel)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/training/saver.py", line 1312, in restore
    err, "a mismatch between the current graph and the graph")
tensorflow.python.framework.errors_impl.InvalidArgumentError: Restoring from checkpoint failed. This is most likely due to a mismatch between the current graph and the graph from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error:

Assign requires shapes of both tensors to match. lhs shape= [2048,84] rhs shape= [2048,20]
	 [[node save/Assign_1 (defined at tools/demo.py:144) ]]

Caused by op 'save/Assign_1', defined at:
  File "tools/demo.py", line 144, in <module>
    saver = tf.train.Saver()
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/training/saver.py", line 832, in __init__
    self.build()
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/training/saver.py", line 844, in build
    self._build(self._filename, build_save=True, build_restore=True)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/training/saver.py", line 881, in _build
    build_save=build_save, build_restore=build_restore)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/training/saver.py", line 513, in _build_internal
    restore_sequentially, reshape)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/training/saver.py", line 354, in _AddRestoreOps
    assign_ops.append(saveable.restore(saveable_tensors, shapes))
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/training/saving/saveable_object_util.py", line 73, in restore
    self.op.get_shape().is_fully_defined())
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/ops/state_ops.py", line 223, in assign
    validate_shape=validate_shape)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/ops/gen_state_ops.py", line 64, in assign
    use_locking=use_locking, name=name)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/op_def_library.py", line 788, in _apply_op_helper
    op_def=op_def)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/util/deprecation.py", line 507, in new_func
    return func(*args, **kwargs)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/ops.py", line 3300, in create_op
    op_def=op_def)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/framework/ops.py", line 1801, in __init__
    self._traceback = tf_stack.extract_stack()
InvalidArgumentError (see above for traceback): Restoring from checkpoint failed. This is most likely due to a mismatch between the current graph and the graph from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error:

Assign requires shapes of both tensors to match. lhs shape= [2048,84] rhs shape= [2048,20]
	 [[node save/Assign_1 (defined at tools/demo.py:144) ]]

原因：

#加载模型
demo.py文件
net.create_architecture("TEST", 21, 
                          tag='default', anchor_scales=[8, 16, 32])
对比test_net.py文件
net.create_architecture("TEST", imdb.num_classes,tag='default',
                          anchor_scales=cfg.ANCHOR_SCALES,
                          anchor_ratios=cfg.ANCHOR_RATIOS)

对比：21 和 imdb.num_classes
可知这个位置的参数代表种类数目，要改成自己的。