yolov3的pytroch实现

yolov3的pytroch实现

  • 1. darknet.py
    • 二级目录
      • 三级目录

https://zhuanlan.zhihu.com/p/49981816

1. darknet.py

总的来说,darknet.py程序包含函数parse_cfg输入 配置文件路径返回一个列表,其中每一个元素为一个字典类型对应于一个要建立的神经网络模块(层),而函数create_modules用来创建网络层级,而Darknet类的forward函数就是实现网络前向传播函数了,还有个load_weights用来导入预训练的网络权重参数。当然,forward函数中需要产生需要的预测输出形式,因此需要变换输出即函数 predict_transform 在文件 util.py 中,我们在 Darknet 类别的 forward 中使用该函数时,将导入该函数。下一篇就要详细注释util.py 了。

二级目录

三级目录

你可能感兴趣的:(神经网络,机器学习,深度学习,pytorch)