Scale layer in case of 2D inputs

e307065c · Dmitry Kurtaev · d329674f · e307065c
Commit e307065c authored Dec 08, 2017 by Dmitry Kurtaev
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

scale_layer.cpp modules/dnn/src/layers/scale_layer.cpp +2 -2

No files found.
--- 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);
                }
            }