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