From 601851cc7e3f82236734d2a2ceaa420bea7faeee Mon Sep 17 00:00:00 2001
From: Aryansh Omray <aryansh.omray@outlook.com>
Date: Wed, 17 Feb 2021 18:49:52 +0530
Subject: [PATCH] TanH darknet and test

---
 modules/dnn/src/darknet/darknet_io.cpp     | 4 ++++
 modules/dnn/test/test_darknet_importer.cpp | 5 +++++
 2 files changed, 9 insertions(+)

diff --git a/modules/dnn/src/darknet/darknet_io.cpp b/modules/dnn/src/darknet/darknet_io.cpp
index f6d71fd6d4..e3c978a8c0 100644
--- a/modules/dnn/src/darknet/darknet_io.cpp
+++ b/modules/dnn/src/darknet/darknet_io.cpp
@@ -241,6 +241,10 @@ namespace cv {
                     {
                         activation_param.type = "Sigmoid";
                     }
+                    else if (type == "tanh")
+                    {
+                        activation_param.type = "TanH";
+                    }
                     else
                     {
                         CV_Error(cv::Error::StsParseError, "Unsupported activation: " + type);
diff --git a/modules/dnn/test/test_darknet_importer.cpp b/modules/dnn/test/test_darknet_importer.cpp
index 0ecec8c49b..00638f83c5 100644
--- a/modules/dnn/test/test_darknet_importer.cpp
+++ b/modules/dnn/test/test_darknet_importer.cpp
@@ -702,6 +702,11 @@ TEST_P(Test_Darknet_layers, mish)
     testDarknetLayer("mish", true);
 }
 
+TEST_P(Test_Darknet_layers, tanh)
+{
+    testDarknetLayer("tanh");
+}
+
 TEST_P(Test_Darknet_layers, avgpool_softmax)
 {
     testDarknetLayer("avgpool_softmax");
-- 
GitLab