Fix Elu import from ONNX

21ed8924 · Dmitry Kurtaev · adf54d41 · 21ed8924
隐藏空白更改
内联并排

Showing with 4 addition and 0 deletion

modules/dnn/src/onnx/onnx_importer.cpp modules/dnn/src/onnx/onnx_importer.cpp +4 -0

未找到文件。
--- 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";