From e307065c8e3c640094ffcef3b8a0b7072d3e426e Mon Sep 17 00:00:00 2001 From: Dmitry Kurtaev Date: Fri, 8 Dec 2017 17:34:59 +0300 Subject: [PATCH] Scale layer in case of 2D inputs --- modules/dnn/src/layers/scale_layer.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/dnn/src/layers/scale_layer.cpp b/modules/dnn/src/layers/scale_layer.cpp index 266dec226a..0ac5209ed4 100644 --- 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(n); float b = hasBias ? blobs[1].at(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); } }