mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-07-26 07:07:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #17624 from dkurt:dnn_optimize_mish

Browse Source
This commit is contained in:
Alexander Alekhin 2020-06-23 18:43:50 +00:00
commit 2e165053ef
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/dnn/src/layers/elementwise_layers.cpp
View File

@ -667,8 +667,11 @@ struct MishFunctor : public BaseFunctor
{ {
for( int i = 0; i < len; i++ ) for( int i = 0; i < len; i++ )
{ {
// Use fast approximation introduced in https://github.com/opencv/opencv/pull/17200
float x = srcptr[i]; float x = srcptr[i];
dstptr[i] = x * tanh(log(1.0f + exp(x))); float eX = exp(std::min(x, 20.f));
float n = (eX + 2) * eX;
dstptr[i] = (x * n) / (n + 2);
} }
} }
} }