Parse repeated values of ConvolutionParameter

2018-04-11 14:38:05 +03:00 · 2018-04-11 14:38:05 +03:00 · 512632e574
commit 512632e574
parent 070ec313f2
1 changed files with 14 additions and 1 deletions
--- a/modules/dnn/src/layers/layers_common.cpp
+++ b/modules/dnn/src/layers/layers_common.cpp
@ -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