提交 8d21c84e 编写于 作者: A Alexander Alekhin

Merge pull request #19474 from aryanshomray:darknet_tanh

......@@ -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);
......
......@@ -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");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册