Fixed potential division by zero.

43529908 · Davis King · 938e9771 · 43529908
Commit 43529908 authored Jul 24, 2016 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 0 deletions

running_gradient.h dlib/statistics/running_gradient.h +9 -0

No files found.
--- a/dlib/statistics/running_gradient.h
+++ b/dlib/statistics/running_gradient.h
@@ -154,6 +154,15 @@ namespace dlib

        static double normal_cfd(double value, double mean, double stddev) 
        {
+            if (stddev == 0)
+            {
+                if (value < mean)
+                    return 0;
+                else if (value > mean)
+                    return 1;
+                else
+                    return 0.5;
+            }
            value = (value-mean)/stddev;
            return 0.5 * erfc(-value / std::sqrt(2.0));
        }