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)
还没有任何评论哟~
