fix darknet-relu bug in darknet_io.cpp
authorzhuqiang00099 <zhuqiang00099@163.com>
Sat, 26 Sep 2020 04:09:30 +0000 (12:09 +0800)
committerAlexander Alekhin <alexander.a.alekhin@gmail.com>
Fri, 2 Oct 2020 06:16:38 +0000 (06:16 +0000)
modules/dnn/src/darknet/darknet_io.cpp

index bc0b413..c745d5f 100644 (file)
@@ -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);