From a968eadbf108d622dff18806251d830984710bb4 Mon Sep 17 00:00:00 2001 From: zhuqiang00099 Date: Sat, 26 Sep 2020 12:09:30 +0800 Subject: [PATCH] fix darknet-relu bug in darknet_io.cpp --- modules/dnn/src/darknet/darknet_io.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/dnn/src/darknet/darknet_io.cpp b/modules/dnn/src/darknet/darknet_io.cpp index bc0b413588..c745d5f036 100644 --- 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(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);