faster-rcnn错误信息 ： tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [21] rhs shape= [2]
发布于2019-08-07 10:11 阅读(1704) 评论(0) 点赞(5) 收藏(0)
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) ]]
我的代码来自 https://github.com/endernewton/tf-faster-rcnn
这是由于类别数目不一致，导致的 ；看错误信息倒数第二行，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])
所属网站分类: 技术文章 > 博客
作者：怪兽爸爸
链接：https://www.pythonheidong.com/blog/article/9681/a30d91a3e9b501b0ee3d/
来源：python黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容：(最多支持255个字符)
---无人问津也好，技不如人也罢，你都要试着安静下来，去做自己该做的事，而不是让内心的烦躁、焦虑，坏掉你本来就不多的热情和定力