Advertisement

Zynqnet Hls code解析

阅读量:

简介:该部分是作者自己实现的卷积加速模块,在HLS中称为testbench模块。

复制代码
    net_CPU = get_network_config();

该函数为网络模型的配置函数,看一下函数定义:

在本研究中首先进行了网络的初始化,在其中采用默认构造函数作为初始化方法。整个网络的最大层数等于其总层数,在本研究中确定的最大层数是27层。论文中记录的最大权重数量被设定为2,528,800个样本数据点,并认为这是对应于权值文件中的全部参数数量。

复制代码
    network_t *net = new network_t(27, 2528800);

主要调用了addLayer()函数为之前初始化的Net的每层添加参数。

复制代码
    void addLayer(network_t *net, layer_t layer) 

输入:待配置网络的指针,某层(结构体)

作用:将某层的配置参数添加到网络中。

层名称 NAME ,输入特征维度为W×H;输入与输出通道数分别为CI和CO;卷积核尺寸 K、填充量 P、步长 S及ReLU激活函数 R;其中S₁指示该模块是否采用了单像素卷积模块(e 1×1),S₂指示是否采用了三像素卷积模块(e 3×3),G P则代表全局池化模块。

首先给出几个静态局部变量:

复制代码
  static int

全部评论 (0)

还没有任何评论哟~