Merge pull request #6395 from Tauranis:master

49b4af1e · Maksim Shabunin · d8629b0f · edb6a0e8 · 49b4af1e
隐藏空白更改
内联并排

Showing with 9 addition and 2 deletion

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

未找到文件。
--- 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]);
-                        data[j] = t;
+                        if(!cvIsInf(data[j]))
+                        {
+                            double t = scale2*(1. - data[j])/(1. + data[j]);
+                            data[j] = t;
+                        }
+                        else
+                        {
+                            data[j] = -scale2;
+                        }
                    }
                }
                break;