fix overflow

29388f80 · fengyuentau · 4792837f · 29388f80
隐藏空白更改
内联并排

Showing with 8 addition and 1 deletion

modules/dnn/src/layers/elementwise_layers.cpp modules/dnn/src/layers/elementwise_layers.cpp +8 -1

未找到文件。
--- a/modules/dnn/src/layers/elementwise_layers.cpp
+++ b/modules/dnn/src/layers/elementwise_layers.cpp
@@ -680,7 +680,14 @@ struct SigmoidFunctor : public BaseDefaultFunctor<SigmoidFunctor>

    inline float calculate(float x) const
    {
-        return 1.f / (1.f + exp(-x));
+        float y;
+        if (x >= 0)
+            y = 1.f / (1.f + exp(-x));
+        else {
+            y = exp(x);
+            y = y / (1 + y);
+        }
+        return y;
    }

 #ifdef HAVE_HALIDE