From 3b8a3f6764397f96edef26a08cd8bf054ab3820c Mon Sep 17 00:00:00 2001 From: nihuini Date: Thu, 10 May 2018 10:48:25 +0800 Subject: [PATCH] convert depthwiseconvolution from yonghenglh6 branch --- tools/caffe/caffe2ncnn.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/caffe/caffe2ncnn.cpp b/tools/caffe/caffe2ncnn.cpp index bf382c83e..1995119bf 100644 --- a/tools/caffe/caffe2ncnn.cpp +++ b/tools/caffe/caffe2ncnn.cpp @@ -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);