faster-rcnn错误信息 : tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [21] rhs s
程序员文章站
2022-05-18 21:14:53
faster-rcnn错误信息 : tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [21] rhs s ......
faster-rcnn错误信息 : tensorflow.python.framework.errors_impl.invalidargumenterror: assign requires shapes of both tensors to match. lhs shape= [21] rhs shape= [2]
1 traceback (most recent call last): 2 file "c:\users\administrator\appdata\local\programs\python\python36\lib\site-packages\tensorflow\python\training\saver.py", line 1276, in restore 3 {self.saver_def.filename_tensor_name: save_path}) 4 file "c:\users\administrator\appdata\local\programs\python\python36\lib\site-packages\tensorflow\python\client\session.py", line 929, in run 5 run_metadata_ptr) 6 file "c:\users\administrator\appdata\local\programs\python\python36\lib\site-packages\tensorflow\python\client\session.py", line 1152, in _run 7 feed_dict_tensor, options, run_metadata) 8 file "c:\users\administrator\appdata\local\programs\python\python36\lib\site-packages\tensorflow\python\client\session.py", line 1328, in _do_run 9 run_metadata) 10 file "c:\users\administrator\appdata\local\programs\python\python36\lib\site-packages\tensorflow\python\client\session.py", line 1348, in _do_call 11 raise type(e)(node_def, op, message) 12 tensorflow.python.framework.errors_impl.invalidargumenterror: assign requires shapes of both tensors to match. lhs shape= [21] rhs shape= [2] 13 [[node save/assign_517 (defined at g:/jianfeng/project/rubblish_det/faster_rcnn/tf-faster-rcnn-master/tools/demo.py:149) ]]
我的代码来自
这是由于类别数目不一致,导致的 ;看错误信息倒数第二行,lhs为21类 rhs为2类
解决方法:在demo.py文件中找到下面两行代码,将 "test"后的21改为 2 因为我的数据类别是 background+单类别=2
1 net.create_architecture("test", 21, 2 tag='default', anchor_scales=[8, 16, 32])
上一篇: 闭包(closure)
下一篇: 1-2 图片批量裁剪
推荐阅读
-
faster-rcnn错误信息 : tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [21] rhs s
-
faster-rcnn错误信息 : tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [21] rhs s