Bug fix for MLP predict for small values to avoid nan responses.

edb6a0e8 · Tauranis · 2f0676fb · edb6a0e8
Commit edb6a0e8 authored Apr 12, 2016 by Tauranis
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 2 deletions

ann_mlp.cpp modules/ml/src/ann_mlp.cpp +9 -2

No files found.
--- a/modules/ml/src/ann_mlp.cpp
+++ b/modules/ml/src/ann_mlp.cpp
@@ -432,8 +432,15 @@ public:
                    double* data = sums.ptr<double>(i);
                    for( j = 0; j < cols; j++ )
                    {
-                        double t = scale2*(1. - data[j])/(1. + data[j]);
+                        if(!cvIsInf(data[j]))
-                        data[j] = t;
+                        {
+                            double t = scale2*(1. - data[j])/(1. + data[j]);
+                            data[j] = t;
+                        }
+                        else
+                        {
+                            data[j] = -scale2;
+                        }
                    }
                }
                break;