From edb6a0e889886c958c908a61a38d8a0953efcf00 Mon Sep 17 00:00:00 2001 From: Tauranis Date: Tue, 12 Apr 2016 08:28:38 -0300 Subject: [PATCH] Bug fix for MLP predict for small values to avoid nan responses. --- modules/ml/src/ann_mlp.cpp | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/modules/ml/src/ann_mlp.cpp b/modules/ml/src/ann_mlp.cpp index 19ee913..8974b6b 100644 --- a/modules/ml/src/ann_mlp.cpp +++ b/modules/ml/src/ann_mlp.cpp @@ -432,8 +432,15 @@ public: double* data = sums.ptr(i); for( j = 0; j < cols; j++ ) { - double t = scale2*(1. - data[j])/(1. + data[j]); - data[j] = t; + if(!cvIsInf(data[j])) + { + double t = scale2*(1. - data[j])/(1. + data[j]); + data[j] = t; + } + else + { + data[j] = -scale2; + } } } break; -- 2.7.4