yolov3 权重转换

这个是blocks：

https://github.com/marvis/pytorch-yolo2/blob/master/darknet.py

  

   

    

     

    
    

     

      import os
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch.nn.functional as F
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      from utils.parse_config import *
     
    
   

    

     

    
    

     

      from utils.utils import *
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      ONNX_EXPORT = False
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def create_modules(module_defs):
     
    
   

    

     

    
    

     
 """
     
    
   

    

     

    
    

     

       Constructs module list of layer blocks from module configuration in module_defs
     
    
   

    

     

    
    

     

       """
     
    
   

    

     

    
    

     

       hyperparams = module_defs.pop(0)
     
    
   

    

     

    
    

     

       output_filters = [int(hyperparams['channels'])]
     
    
   

    

     

    
    

     

       module_list = nn.ModuleList()
     
    
   

    

     

    
    

     

       yolo_layer_count = 0
     
    
   

    

     

    
    

     
 for i, module_def in enumerate(module_defs):
     
    
   

    

     

    
    

     

       modules = nn.Sequential()
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 if module_def['type'] == 'convolutional':
     
    
   

    

     

    
    

     

       bn = int(module_def['batch_normalize'])
     
    
   

    

     

    
    

     

       filters = int(module_def['filters'])
     
    
   

    

     

    
    

     

       kernel_size = int
     
    
  
 

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

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