mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-20 18:10:51 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

mish_functor_update

Browse Source
This commit is contained in:
Sinitsina 2020-07-22 21:26:27 +03:00 committed by Dmitry Kurtaev
parent ef1690ef45
commit 0ac2f0e04c
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -669,12 +669,17 @@ struct MishFunctor : public BaseFunctor
{ {
// Use fast approximation introduced in https://github.com/opencv/opencv/pull/17200 // Use fast approximation introduced in https://github.com/opencv/opencv/pull/17200
float x = srcptr[i]; float x = srcptr[i];
float eX = exp(std::min(x, 20.f)); if (x >= 8.f)
dstptr[i] = x;
else
{
float eX = exp(x);
float n = (eX + 2) * eX; float n = (eX + 2) * eX;
dstptr[i] = (x * n) / (n + 2); dstptr[i] = (x * n) / (n + 2);
} }
} }
} }
}
#ifdef HAVE_OPENCL #ifdef HAVE_OPENCL
bool applyOCL(InputArrayOfArrays inps, OutputArrayOfArrays outs, OutputArrayOfArrays internals) bool applyOCL(InputArrayOfArrays inps, OutputArrayOfArrays outs, OutputArrayOfArrays internals)