文档
注册
评分
提单
论坛
小AI

Convolution(DepthwiseConv,ConvolutionDepthwise)

输入

  • x:
    • 是否必填: 必填
    • 数据类型: float16
    • 参数解释: 对输入x做卷积
    • 规格限制: 无
  • filter:
    • 是否必填: 必填
    • 数据类型: float16
    • 参数解释: 卷积核,DepthwiseConv和ConvolutionDepthwise都通过转换为Convolution算子来使用,例如ConvolutionDepthwise算子:
      layer {
         name: "resx1_conv2"
         type: "ConvolutionDepthwise"
         bottom: "resx1_conv1"
         top: "resx1_conv2"
         convolution_param {
           num_output: 54
           kernel_size: 3
           stride: 2
           pad: 1
           bias_term: false
           weight_filler {
             type: "msra"
           }
         }
       }

      需要修改为:

      layer {
         name: "resx1_conv2"
         type: "Convolution"
         bottom: "resx1_conv1"
         top: "resx1_conv2"
         convolution_param {
           num_output: 54
           group: 54
           kernel_size: 3
           stride: 2
           pad: 1
           bias_term: false
           weight_filler {
             type: "msra"
           }
         }
       }

      如果DepthwiseConv和ConvolutionDepthwise中已携带group属性,则只修改type为Convolution即可;如果没有携带group属性,则需添加group属性: group==num_output(group、out_channel和input_channel必须相同)

      修改后的模型文件(.prototxt)和权重文件(.caffemodel)的op name、op type必须保持名称一致(包括大小写)。

    • 规格限制:
      1. 如果out_w==out_h==1: feature_w==feature_h==filter_w==filter_h,且取值范围在1~11
      2. 不支持输出的W==1,输出的H不等于1的场景
      3. 如果group==Channel,则要求 Filter_h*filter_w*32+feature_w*(31*stride_h+filter_w)<=32768,且满足Filter_w,filter_h配置范围在1~255
      4. 其它场景: Filter_W和Filter_h配置范围在1~255
  • bias:
    • 是否必填: 非必填
    • 数据类型: float16
    • 参数解释: bias,可以为None(表示不需要bias)
    • 规格限制: 无

属性

  • pad:
    • 是否必填: 非必填
    • 数据类型: ListInt
    • 参数解释: 高和宽轴开始和结束的Padding,pad和pad_h/pad_w不能同时提供,默认值为0;pad的List长度最大为2
    • 规格限制: pad<filter.size
  • pad_h:
    • 是否必填: 非必填
    • 数据类型: int
    • 参数解释: 高轴开始和结束的Padding,pad和pad_h/pad_w不能同时提供,默认值为0;pad的List长度最大为2
    • 规格限制: pad<filter.size
  • pad_w:
    • 是否必填: 非必填
    • 数据类型: int
    • 参数解释: 宽轴开始和结束的Padding,pad和pad_h/pad_w不能同时提供,默认值为0;pad的List长度最大为2
    • 规格限制: pad<filter.size
  • stride:
    • 是否必填: 非必填
    • 数据类型: ListInt
    • 参数解释: 高和宽轴的stride.stride和stride_h/stride_w不能同时提供,默认值为1;stride的List长度最大为2
    • 规格限制: 1<=stride<=63
  • stride_h:
    • 是否必填: 非必填
    • 数据类型: int
    • 参数解释: 高轴的stride.stride和stride_h/stride_w不能同时提供,默认值为1;stride的List长度最大为2
    • 规格限制: 1<=stride<=63
  • stride_w:
    • 是否必填: 非必填
    • 数据类型: int
    • 参数解释: 宽轴的stride.stride和stride_h/stride_w不能同时提供,默认值为1;stride的List长度最大为2
    • 规格限制: 1<=stride<=63
  • dilation:
    • 是否必填: 非必填
    • 数据类型: ListInt
    • 参数解释: Filter的高和宽轴的放大系数,List长度最大为2,默认值为1
    • 规格限制: 支持1~255,配置后(kernel-1)*dilation+1<256
  • group:
    • 是否必填: 非必填
    • 数据类型: int
    • 参数解释: 无
    • 规格限制: group能被channel整除

输出

y:
  • 是否必填: 必填
  • 数据类型: float16
  • 参数解释: 无
  • 规格限制: 无
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词