Merge pull request #12333 from wzw-intel:prior_box

7 years ago · 8a8d07fb17
parent 087e9308ec ca51bbb7ff
commit 8a8d07fb17
1 changed files with 2 additions and 2 deletions
--- a/modules/dnn/src/layers/prior_box_layer.cpp
+++ b/modules/dnn/src/layers/prior_box_layer.cpp
@ -453,8 +453,8 @@ public:
        outputPtr = outputs[0].ptr<float>(0, 1);
        if(_variance.size() == 1)
        {
-            Mat secondChannel(outputs[0].size[2], outputs[0].size[3], CV_32F, outputPtr);
-            secondChannel.setTo(Scalar(_variance[0]));
+            Mat secondChannel(1, outputs[0].size[2], CV_32F, outputPtr);
+            secondChannel.setTo(Scalar::all(_variance[0]));
        }
        else
        {