From 6dfe2ddc21cfc3216a3f9483b56eace916eb679f Mon Sep 17 00:00:00 2001 From: Diego Date: Thu, 4 Jul 2019 07:56:00 +0200 Subject: [PATCH] Merge pull request #14959 from dvd42:onnx_clip * onnx clip operation --- modules/dnn/src/onnx/onnx_importer.cpp | 7 +++++++ modules/dnn/test/test_onnx_importer.cpp | 5 +++++ 2 files changed, 12 insertions(+) diff --git a/modules/dnn/src/onnx/onnx_importer.cpp b/modules/dnn/src/onnx/onnx_importer.cpp index 696588c83a..b9fb0be624 100644 --- a/modules/dnn/src/onnx/onnx_importer.cpp +++ b/modules/dnn/src/onnx/onnx_importer.cpp @@ -530,6 +530,13 @@ void ONNXImporter::populateNet(Net dstNet) layerParams.type = "Power"; } } + else if (layer_type == "Clip") + { + layerParams.type = "ReLU6"; + replaceLayerParam(layerParams, "min", "min_value"); + replaceLayerParam(layerParams, "max", "max_value"); + + } else if (layer_type == "LeakyRelu") { layerParams.type = "ReLU"; diff --git a/modules/dnn/test/test_onnx_importer.cpp b/modules/dnn/test/test_onnx_importer.cpp index c99b8cf431..186239494f 100644 --- a/modules/dnn/test/test_onnx_importer.cpp +++ b/modules/dnn/test/test_onnx_importer.cpp @@ -136,6 +136,11 @@ TEST_P(Test_ONNX_layers, ReLU) testONNXModels("ReLU"); } +TEST_P(Test_ONNX_layers, Clip) +{ + testONNXModels("clip", npy); +} + TEST_P(Test_ONNX_layers, MaxPooling_Sigmoid) { testONNXModels("maxpooling_sigmoid");