mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-08-04 21:36:35 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix darknet-relu bug in darknet_io.cpp

Browse Source
This commit is contained in:
zhuqiang00099 2020-09-26 12:09:30 +08:00 committed by Alexander Alekhin
parent 969b55036f
commit a968eadbf1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/dnn/src/darknet/darknet_io.cpp
View File

@ -984,8 +984,8 @@ namespace cv {
} }
std::string activation = getParam<std::string>(layer_params, "activation", "linear"); std::string activation = getParam<std::string>(layer_params, "activation", "linear");
if(activation == "leaky" || activation == "swish" || activation == "mish" || activation == "logistic") if (activation != "linear")
++cv_layers_counter; // For ReLU, Swish, Mish, Sigmoid ++cv_layers_counter; // For ReLU, Swish, Mish, Sigmoid, etc
if(!darknet_layers_counter) if(!darknet_layers_counter)
tensor_shape.resize(1); tensor_shape.resize(1);