generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #5854 from Dikay900:em_predict_fix

Browse Source
This commit is contained in:
Alexander Alekhin 2015-12-22 15:15:53 +00:00
commit f550ddb3fd
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/ml/src/em.cpp
View File

@ -198,7 +198,7 @@ public:
sample.convertTo(tmp, CV_64FC1); sample.convertTo(tmp, CV_64FC1);
sample = tmp; sample = tmp;
} }
sample.reshape(1, 1); sample = sample.reshape(1, 1);
Mat probs; Mat probs;
if( _probs.needed() ) if( _probs.needed() )
@ -342,7 +342,7 @@ public:
if(weights0 && (startStep == START_E_STEP && covs0)) if(weights0 && (startStep == START_E_STEP && covs0))
{ {
weights0->convertTo(weights, CV_64FC1); weights0->convertTo(weights, CV_64FC1);
weights.reshape(1,1); weights = weights.reshape(1,1);
preprocessProbability(weights); preprocessProbability(weights);
} }