third_party/opencv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add a test for grouped deconvolution from ONNX

Browse Source
This commit is contained in:
Dmitry Kurtaev 2019-02-21 08:54:35 +03:00
parent 5c7fe0fe05
commit bfd663c281
2 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/dnn/src/layers/convolution_layer.cpp
View File

@ -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;
}

1
modules/dnn/test/test_onnx_importer.cpp
View File

@ -72,6 +72,7 @@ TEST_P(Test_ONNX_layers, Deconvolution)
{
testONNXModels("deconvolution");
testONNXModels("two_deconvolution");
testONNXModels("deconvolution_group");
}
TEST_P(Test_ONNX_layers, Dropout)