当前位置： 移动技术网 > IT编程>脚本编程>Python > 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

2019年07月16日  | 移动技术网IT编程  | 我要评论

尚亚堂,智通竞答,天河部落





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




 


                    

                        

                            
                        

                    

                    


                    
                    

                        
    
                            
  • 
                                
                            
    • 
                        

                    


                    
                    

                    

                        
您可能感兴趣的文章:

                        
                    

                    

                        如对本文有疑问，请在下面进行留言讨论，广大热心网友会与你互动！！
                        点击进行留言回复
                    

                

                 
                

                    
相关文章:

                    
                

               
                       
                             

                

                    
                    

                        
                        
                        

                        

                            

                                
                                    
                                
                                 
                                    
                                
                            

                            

                                验证码：
                                
                                    
                                    
                                    
                                
                            

                        

                        
                    

                    
                    

                        
                        
                        
                    

                

            

        

        
        
 
            

                

                    
                    

                    
                


                

                    

                        

                            
最近更新的文章

                            

                                
                            

                        

                    

                


                

                    
                    

                        
                    

                


                

                

                    

                        

                            
大家感兴趣的文章

                            

                                
                            

                        

                    

                

                

                    
                    

                        
                    

                


            

        

    



 


 
 

     
    

        

移动技术网