Merge pull request #5657 from tianzhi0549:fixed-the-type-inconsistent-with-document

9 years ago · bfe5ed23cd
parent a0f6cd8df5 f0434d60b0
commit bfe5ed23cd
1 changed files with 2 additions and 2 deletions
--- a/modules/ml/src/em.cpp
+++ b/modules/ml/src/em.cpp
@ -161,7 +161,7 @@ public:
    {
        bool needprobs = _outputs.needed();
        Mat samples = _inputs.getMat(), probs, probsrow;
-        int ptype = CV_32F;
+        int ptype = CV_64F;
        float firstres = 0.f;
        int i, nsamples = samples.rows;

@ -187,7 +187,7 @@ public:

    Vec2d predict2(InputArray _sample, OutputArray _probs) const
    {
-        int ptype = CV_32F;
+        int ptype = CV_64F;
        Mat sample = _sample.getMat();
        CV_Assert(isTrained());