faster-rcnn錯誤信息 ： tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shap

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])