ml: fix NormalBayesClassifier bulk prediction(#5911)

2016-01-04 11:47:08 +01:00
parent fade402899
commit 5afd0e211e
2 changed files with 45 additions and 1 deletions
--- a/modules/ml/src/nbayes.cpp
+++ b/modules/ml/src/nbayes.cpp
@@ -236,7 +236,7 @@ public:
            if (results_prob)
            {
                rptype = results_prob->type();
-                rpstep = results_prob->isContinuous() ? 1 : results_prob->step/results_prob->elemSize();
+                rpstep = results_prob->isContinuous() ? results_prob->cols : results_prob->step/results_prob->elemSize();
            }
            // allocate memory and initializing headers for calculating
            cv::AutoBuffer<double> _buffer(nvars*2);