From d0504c95f47c21302b707d382786d1ac30eebb53 Mon Sep 17 00:00:00 2001 From: Dmitry Kurtaev Date: Wed, 9 Jan 2019 13:10:19 +0300 Subject: [PATCH] Add a text message for Convolution layer's input channels check --- modules/dnn/src/layers/convolution_layer.cpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/modules/dnn/src/layers/convolution_layer.cpp b/modules/dnn/src/layers/convolution_layer.cpp index 9daceb5c13..5dbd5ba895 100644 --- a/modules/dnn/src/layers/convolution_layer.cpp +++ b/modules/dnn/src/layers/convolution_layer.cpp @@ -259,6 +259,9 @@ public: } int ngroups = inpCn / blobs[0].size[1]; + if (ngroups == 0 || ngroups * blobs[0].size[1] != inpCn) + CV_Error(Error::StsError, format("Number of input channels should " + "be multiple of %d but got %d", blobs[0].size[1], inpCn)); CV_Assert(ngroups > 0 && inpCn % ngroups == 0 && outCn % ngroups == 0); int dims[] = {inputs[0][0], outCn, out.height, out.width};