From 05db8784ae43557a7958a752738d989372ece211 Mon Sep 17 00:00:00 2001 From: Smirnov Egor Date: Mon, 29 Nov 2021 19:56:23 +0300 Subject: [PATCH] fix Clip, LeakyReLU, LRN, Split defaults --- modules/dnn/src/layers/lrn_layer.cpp | 2 +- modules/dnn/src/onnx/onnx_importer.cpp | 7 ++++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/modules/dnn/src/layers/lrn_layer.cpp b/modules/dnn/src/layers/lrn_layer.cpp index 434ba5ccbe..65cd160dfe 100644 --- a/modules/dnn/src/layers/lrn_layer.cpp +++ b/modules/dnn/src/layers/lrn_layer.cpp @@ -80,7 +80,7 @@ public: if (size % 2 != 1 || size <= 0) CV_Error(Error::StsBadArg, "LRN layer supports only positive odd values for local_size"); - alpha = params.get("alpha", 1); + alpha = params.get("alpha", 0.0001); beta = params.get("beta", 0.75); bias = params.get("bias", 1); normBySize = params.get("norm_by_size", true); diff --git a/modules/dnn/src/onnx/onnx_importer.cpp b/modules/dnn/src/onnx/onnx_importer.cpp index d70a81b9ad..f382d4637e 100644 --- a/modules/dnn/src/onnx/onnx_importer.cpp +++ b/modules/dnn/src/onnx/onnx_importer.cpp @@ -920,6 +920,7 @@ void ONNXImporter::parseSplit(LayerParams& layerParams, const opencv_onnx::NodeP layerParams.set("num_split", node_proto.output_size()); } layerParams.type = "Slice"; + layerParams.set("axis", layerParams.get("axis", 0)); addLayer(layerParams, node_proto); } @@ -1184,15 +1185,15 @@ void ONNXImporter::parseImageScaler(LayerParams& layerParams, const opencv_onnx: void ONNXImporter::parseClip(LayerParams& layerParams, const opencv_onnx::NodeProto& node_proto) { layerParams.type = "ReLU6"; - replaceLayerParam(layerParams, "min", "min_value"); - replaceLayerParam(layerParams, "max", "max_value"); + layerParams.set("min_value", layerParams.get("min", -FLT_MAX)); + layerParams.set("max_value", layerParams.get("max", FLT_MAX)); addLayer(layerParams, node_proto); } void ONNXImporter::parseLeakyRelu(LayerParams& layerParams, const opencv_onnx::NodeProto& node_proto) { layerParams.type = "ReLU"; - replaceLayerParam(layerParams, "alpha", "negative_slope"); + layerParams.set("negative_slope", layerParams.get("alpha", 0.01)); addLayer(layerParams, node_proto); } -- GitLab