Parse repeated values of ConvolutionParameter
authorDmitry Kurtaev <dmitry.kurtaev+github@gmail.com>
Wed, 11 Apr 2018 11:38:05 +0000 (14:38 +0300)
committerDmitry Kurtaev <dmitry.kurtaev+github@gmail.com>
Wed, 11 Apr 2018 11:38:05 +0000 (14:38 +0300)
modules/dnn/src/layers/layers_common.cpp

index f854d91..bf5834c 100644 (file)
@@ -77,7 +77,20 @@ bool getParameter(const LayerParams &params, const std::string& nameBase, const
     {
         if (params.has(nameAll_))
         {
-            parameterH = parameterW = params.get<int>(nameAll_);
+            DictValue param = params.get(nameAll_);
+            parameterH = param.get<int>(0);
+            if (param.size() == 1)
+            {
+                parameterW = parameterH;
+            }
+            else if (param.size() == 2)
+            {
+                parameterW = param.get<int>(1);
+            }
+            else
+            {
+                return false;
+            }
             return true;
         }
         else