faster-rcnn错误信息 ： tensorflow.python.fram…
2019-07-24 09:24:35来源：博客园 阅读 ()
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.cnblogs.com/bob-jianfeng/p/11125696.html
如有疑问请与原作者联系
标签：
版权申明：本站文章部分自网络，如有侵权，请联系：west999com@outlook.com
特别注意：本站所有转载文章言论不代表本站观点，本站所提供的摄影照片，插画，设计作品，如需使用，请与原作者联系，版权归原作者所有
- 使用scrapy框架爬取全书网书籍信息。 2019-08-13
- pycharm查看函数用法，参数信息的设置方法 2019-07-24
- [Python学习]错误篇二：切换当前工作目录时出错——FileNotF 2019-07-24
- mysql-python 安装错误: Cannot open include file:  2019-07-24
- 读文件时出现这个错误 'utf-8' codec can&a 2019-07-24
IDC资讯： 主机资讯 注册资讯 托管资讯 vps资讯 网站建设
网站运营： 建站经验 策划盈利 搜索优化 网站推广 免费资源
网络编程： Asp.Net编程 Asp编程 Php编程 Xml编程 Access Mssql Mysql 其它
服务器技术： Web服务器 Ftp服务器 Mail服务器 Dns服务器 安全防护
软件技巧： 其它软件 Word Excel Powerpoint Ghost Vista QQ空间 QQ FlashGet 迅雷
网页制作： FrontPages Dreamweaver Javascript css photoshop fireworks Flash
