yolov3 权重转换

最新推荐文章于 2024-08-08 08:30:30 发布

AI算法网奇

最新推荐文章于 2024-08-08 08:30:30 发布

阅读量5.2k

点赞数

分类专栏： pytorch知识宝典深度学习宝典

本文链接：https://blog.csdn.net/jacke121/article/details/92011159

版权

深度学习宝典同时被 2 个专栏收录

985 篇文章 713 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

pytorch知识宝典

499 篇文章 226 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文档介绍了如何将YoloV3的权重文件转换到其他框架，引用了GitHub上的darknet.py源码作为转换参考。

摘要由CSDN通过智能技术生成

这个是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&