Browse Source

convert depthwiseconvolution from yonghenglh6 branch

tags/20180704
nihuini 8 years ago
parent
commit
3b8a3f6764
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      tools/caffe/caffe2ncnn.cpp

+ 2
- 2
tools/caffe/caffe2ncnn.cpp View File

@@ -332,7 +332,7 @@ int main(int argc, char** argv)
else
fprintf(pp, "%-16s", "Convolution");
}
else if (layer.type() == "ConvolutionDepthwise")
else if (layer.type() == "ConvolutionDepthwise" || layer.type() == "DepthwiseConvolution")
{
fprintf(pp, "%-16s", "ConvolutionDepthWise");
}
@@ -462,7 +462,7 @@ int main(int argc, char** argv)
int dim = concat_param.axis() - 1;
fprintf(pp, " 0=%d", dim);
}
else if (layer.type() == "Convolution" || layer.type() == "ConvolutionDepthwise")
else if (layer.type() == "Convolution" || layer.type() == "ConvolutionDepthwise" || layer.type() == "DepthwiseConvolution")
{
const caffe::LayerParameter& binlayer = net.layer(netidx);



Loading…
Cancel
Save