yolo网络参数的解析与存储

参考:
yolo网络分析
yolo v3论文解析

yolo网络结构定义的cfg文件
  1. yolo中的网络采用和caffe差不多的方式。都是通过一个顺序堆叠layer对神经网络结构进行定义的文件。区别在于:yolo采用自己定义的cfg文件格式,需要自己实现解析功能。而caffe无需自己实现解析文件的功能,但缺点是对第三方库依赖严重,在编译常出现无法链接的错误。
  2. cfg文件的格式可以总结如下:
[net]
#下面对net的参数进行配置
#yolo对net求解器的参数也放在了这里
[conv]
#一些conv层的参数描述
[maxpool]
#池化层的参数描述

……
#顺序堆叠的其他层的参数描述
网络结构解析器parser
  1. 具体的解析见parser.c文件。先以convolutional_layer parse_convolutional(list *options, size_params params)函数为例,看darknet是如何完成对卷基层参数的解析的。这个函数接受一个list类型的变量,darknet将堆叠起来的这些层描述抽象成链表。另一个参数是size_params类型的params变量,它指示了该层上一层的参数状况。具体定义如下:
typedef struct size_params{
    int batch;
    int inputs;
    int h;
    int w;
    int c;
    int index;
    int time_steps;
    network net;
} size_params;

这样在构建该层卷积层时,就能够知道上一层的输入维度等信息,方便做一些参数检查和layer初始化的工作。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值