Merge pull request #11295 from dkurt:dnn_repeated_conv_params

2026-07-29 23:33:05 +04:00 · 2018-04-11 15:25:24 +00:00
parent 59221296c0 512632e574
commit 0b9d075958
1 changed files with 14 additions and 1 deletions
@@ -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