Scale layer in case of 2D inputs
authorDmitry Kurtaev <dmitry.kurtaev+github@gmail.com>
Fri, 8 Dec 2017 14:34:59 +0000 (17:34 +0300)
committerDmitry Kurtaev <dmitry.kurtaev+github@gmail.com>
Fri, 8 Dec 2017 14:34:59 +0000 (17:34 +0300)
modules/dnn/src/layers/scale_layer.cpp

index 266dec2..0ac5209 100644 (file)
@@ -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);
                 }
             }