mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-01-19 06:53:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix Elu import from ONNX

Browse Source
This commit is contained in:
Dmitry Kurtaev 2020-04-08 12:09:45 +03:00
parent adf54d41d5
commit 21ed892489
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/dnn/src/onnx/onnx_importer.cpp
View File

@ -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";