caffe 模型中的形状大小不规则
Irregular size in shapes in caffe model
当我尝试像这样在 caffe 中制作自己的模型时:
n.data = L.Input(input_param={'shape':{'dim':[1,1,64,64]}})
n.conv1 = L.Convolution(n.data, kernel_size=5,
num_output=16, pad=1, weight_filler=dict(type='xavier'))
n.elu1 = L.ELU(n.conv1, in_place=True)
n.scale1 = L.Scale(n.elu1, bias_term=False, in_place=True)
我得到了 62x62x16 的输出形状,但正确的做法是得到 64x64x16 之一,我的代码有错误吗?
Output_size 是 ((Input_size+2(填充)-kernel_size)/步幅)+1
Input_size是64,kernel_size是5,padding是1
所以 ((64+2-5)/1)+1 = 62
您需要更改填充或kernel_size。
当我尝试像这样在 caffe 中制作自己的模型时:
n.data = L.Input(input_param={'shape':{'dim':[1,1,64,64]}})
n.conv1 = L.Convolution(n.data, kernel_size=5,
num_output=16, pad=1, weight_filler=dict(type='xavier'))
n.elu1 = L.ELU(n.conv1, in_place=True)
n.scale1 = L.Scale(n.elu1, bias_term=False, in_place=True)
我得到了 62x62x16 的输出形状,但正确的做法是得到 64x64x16 之一,我的代码有错误吗?
Output_size 是 ((Input_size+2(填充)-kernel_size)/步幅)+1
Input_size是64,kernel_size是5,padding是1
所以 ((64+2-5)/1)+1 = 62
您需要更改填充或kernel_size。