torch.load invalid load key, ‘\x00‘

torch加载yolov3.weights，报错：

torch.load invalid load key, '\x00'

  

   

    

     

    
    

     

      cfg='cfg/yolov3.cfg'
     
    
   

    

     

    
    

     

      model = Darknet(cfg, img_size)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 # Load weights
     
    
   

    

     

    
    

     
 if weights.endswith('.pt'):  # pytorch format
     
    
   

    

     

    
    

     

       model.load_state_dict(torch.load(weights, map_location=device)['model'])
     
    
   

    

     

    
    

     
 else:  # darknet format
     
    
   

    

     

    
    

     

       _ = load_darknet_weights(model, weights)
     
    
  
 

可以尝试的方法：用numpy.fromfile来读试试，能解决部分问题:

加载权重，保存权重，转换权重

  

   

    

     

    
    

     

      def load_darknet_weights(self, weights, cutoff=-1):
     
    
   

    

     

    
    

     
 # Parses and loads the weights stored in 'weights'
     
    
   

    

     

    
    

     
 # cutoff: save layers between 0 and cutoff (if cutoff = -1 all are saved)
     
    
   

    

     

    
    

     

       weights_file = weights.split(os.sep)[-1]
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 # Try to download weights if not available locally
     
    
   

    

     

    
    

     
 if not os.path.isfile(weights):
     
    
   

    

     

    
    

     
 try:
     
    
   

    

     

    
    

     

       os.system(
     
    
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/92008694