diff --git a/tools/caffe/caffe2ncnn.cpp b/tools/caffe/caffe2ncnn.cpp index 7baf5f215..5572958cb 100644 --- a/tools/caffe/caffe2ncnn.cpp +++ b/tools/caffe/caffe2ncnn.cpp @@ -638,7 +638,7 @@ int main(int argc, char** argv) fprintf(pp, " 6=%d", weight_blob.data_size()); int num_group = 1; - if (layer.type() == "ConvolutionDepthwise") + if (layer.type() == "ConvolutionDepthwise" || layer.type() == "DepthwiseConvolution") { num_group = convolution_param.num_output(); }