From bfd663c281f684fddf72db517f5d6f2b1f5e0cff Mon Sep 17 00:00:00 2001 From: Dmitry Kurtaev Date: Thu, 21 Feb 2019 08:54:35 +0300 Subject: [PATCH] Add a test for grouped deconvolution from ONNX --- modules/dnn/src/layers/convolution_layer.cpp | 6 +----- modules/dnn/test/test_onnx_importer.cpp | 1 + 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/modules/dnn/src/layers/convolution_layer.cpp b/modules/dnn/src/layers/convolution_layer.cpp index 60611b52b2..453892ec45 100644 --- a/modules/dnn/src/layers/convolution_layer.cpp +++ b/modules/dnn/src/layers/convolution_layer.cpp @@ -1217,12 +1217,8 @@ public: int dims[] = {inputs[0][0], outCn, outH, outW}; outputs.resize(inputs.size(), shape(dims, 4)); - internals.push_back(MatShape()); if (!is1x1()) - internals[0] = computeColRowShape(inputs[0], outputs[0]); - - if (hasBias()) - internals.push_back(shape(1, outH*outW)); + internals.push_back(computeColRowShape(inputs[0], outputs[0])); return false; } diff --git a/modules/dnn/test/test_onnx_importer.cpp b/modules/dnn/test/test_onnx_importer.cpp index 217ef34421..5bd517433f 100644 --- a/modules/dnn/test/test_onnx_importer.cpp +++ b/modules/dnn/test/test_onnx_importer.cpp @@ -72,6 +72,7 @@ TEST_P(Test_ONNX_layers, Deconvolution) { testONNXModels("deconvolution"); testONNXModels("two_deconvolution"); + testONNXModels("deconvolution_group"); } TEST_P(Test_ONNX_layers, Dropout) -- GitLab