fix darknet-relu bug in darknet_io.cpp

5 years ago · a968eadbf1
parent 969b55036f
commit a968eadbf1
1 changed files with 2 additions and 2 deletions
--- a/modules/dnn/src/darknet/darknet_io.cpp
+++ b/modules/dnn/src/darknet/darknet_io.cpp
@ -984,8 +984,8 @@ namespace cv {
                    }

                    std::string activation = getParam<std::string>(layer_params, "activation", "linear");
-                    if(activation == "leaky" || activation == "swish" || activation == "mish" || activation == "logistic")
-                        ++cv_layers_counter;  // For ReLU, Swish, Mish, Sigmoid
+                    if (activation != "linear")
+                        ++cv_layers_counter;  // For ReLU, Swish, Mish, Sigmoid, etc

                    if(!darknet_layers_counter)
                        tensor_shape.resize(1);