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])
?
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索：z360901061
微信掃一掃加我為好友
QQ號聯系： 360901061
您的支持是博主寫作最大的動力，如果您喜歡我的文章，感覺我的文章對您有幫助，請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧，狠狠點擊下面給點支持吧，站長非常感激您！手機微信長按不能支付解決辦法：請將微信支付二維碼保存到相冊，切換到微信，然后點擊微信右上角掃一掃功能，選擇支付二維碼完成支付。
【本文對您有幫助就好】元
