From b58a8729c246090a3c0f27f9f124ecb04df535e2 Mon Sep 17 00:00:00 2001 From: Christopher Gundler Date: Fri, 16 Nov 2018 20:50:40 +0100 Subject: [PATCH] Merge pull request #13131 from Christopher22:add_transposedConv_onnx * Add support for ConvTranspose when parsing ONNX. * Add support for ConvTranspose when parsing ONNX. * Add test for Deconvolution --- modules/dnn/src/onnx/onnx_importer.cpp | 10 ++++++++++ modules/dnn/test/test_onnx_importer.cpp | 6 ++++++ 2 files changed, 16 insertions(+) diff --git a/modules/dnn/src/onnx/onnx_importer.cpp b/modules/dnn/src/onnx/onnx_importer.cpp index ac48d47b55..f1c6f43271 100644 --- a/modules/dnn/src/onnx/onnx_importer.cpp +++ b/modules/dnn/src/onnx/onnx_importer.cpp @@ -508,6 +508,16 @@ void ONNXImporter::populateNet(Net dstNet) layerParams.set("num_output", layerParams.blobs[0].size[0]); layerParams.set("bias_term", node_proto.input_size() == 3); } + else if (layer_type == "ConvTranspose") + { + CV_Assert(node_proto.input_size() >= 2); + layerParams.type = "Deconvolution"; + for (int j = 1; j < node_proto.input_size(); j++) { + layerParams.blobs.push_back(getBlob(node_proto, constBlobs, j)); + } + layerParams.set("num_output", layerParams.blobs[0].size[1]); + layerParams.set("bias_term", node_proto.input_size() == 3); + } else if (layer_type == "Transpose") { layerParams.type = "Permute"; diff --git a/modules/dnn/test/test_onnx_importer.cpp b/modules/dnn/test/test_onnx_importer.cpp index d603dd9acd..61b06cc7cf 100644 --- a/modules/dnn/test/test_onnx_importer.cpp +++ b/modules/dnn/test/test_onnx_importer.cpp @@ -68,6 +68,12 @@ TEST_P(Test_ONNX_layers, Convolution) testONNXModels("two_convolution"); } +TEST_P(Test_ONNX_layers, Deconvolution) +{ + testONNXModels("deconvolution"); + testONNXModels("two_deconvolution"); +} + TEST_P(Test_ONNX_layers, Dropout) { testONNXModels("dropout"); -- GitLab