Merge pull request #19474 from aryanshomray:darknet_tanh

pull/19559/head
Alexander Alekhin 4 years ago
commit 8d21c84e7d
  1. 4
      modules/dnn/src/darknet/darknet_io.cpp
  2. 5
      modules/dnn/test/test_darknet_importer.cpp

@ -241,6 +241,10 @@ namespace cv {
{ {
activation_param.type = "Sigmoid"; activation_param.type = "Sigmoid";
} }
else if (type == "tanh")
{
activation_param.type = "TanH";
}
else else
{ {
CV_Error(cv::Error::StsParseError, "Unsupported activation: " + type); CV_Error(cv::Error::StsParseError, "Unsupported activation: " + type);

@ -702,6 +702,11 @@ TEST_P(Test_Darknet_layers, mish)
testDarknetLayer("mish", true); testDarknetLayer("mish", true);
} }
TEST_P(Test_Darknet_layers, tanh)
{
testDarknetLayer("tanh");
}
TEST_P(Test_Darknet_layers, avgpool_softmax) TEST_P(Test_Darknet_layers, avgpool_softmax)
{ {
testDarknetLayer("avgpool_softmax"); testDarknetLayer("avgpool_softmax");

Loading…
Cancel
Save