Scale layer in case of 2D inputs

7 years ago · e307065c8e
parent d329674f68
commit e307065c8e
1 changed files with 2 additions and 2 deletions
--- a/modules/dnn/src/layers/scale_layer.cpp
+++ b/modules/dnn/src/layers/scale_layer.cpp
@ -74,8 +74,8 @@ public:
                {
                    float w = blobs[0].at<float>(n);
                    float b = hasBias ? blobs[1].at<float>(n) : 0;
-                    Mat outBlobPlane = getPlane(outBlob, cn, n);
-                    Mat inpBlobPlane = getPlane(inpBlob, cn, n);
+                    Mat outBlobPlane = slice(outBlob, cn, n);
+                    Mat inpBlobPlane = slice(inpBlob, cn, n);
                    inpBlobPlane.convertTo(outBlobPlane, CV_32F, w, b);
                }
            }