From d5df2301cb7f58a34db37e9efb44de157d2b0c68 Mon Sep 17 00:00:00 2001
From: BUG1989 <248857878@qq.com>
Date: Thu, 23 Aug 2018 18:12:44 +0800
Subject: [PATCH]  fix the caffe2ncnn that "DepthwiseConvolution" loss group
 num param (#540)

* fix the caffe2ncnn  bug that "DepthwiseConvolution" loss group num param
---
 tools/caffe/caffe2ncnn.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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();
             }