Fix Elu import from ONNX

5 years ago · 21ed892489
parent adf54d41d5
commit 21ed892489
1 changed files with 4 additions and 0 deletions
--- a/modules/dnn/src/onnx/onnx_importer.cpp
+++ b/modules/dnn/src/onnx/onnx_importer.cpp
@ -777,6 +777,10 @@ void ONNXImporter::populateNet(Net dstNet)
        {
            layerParams.type = "ReLU";
        }
+        else if (layer_type == "Elu")
+        {
+            layerParams.type = "ELU";
+        }
        else if (layer_type == "PRelu")
        {
            layerParams.type = "PReLU";