Skip to content

O
opencv
Commit 2bacd8b7 authored by berak's avatar berak
Browse files
Options

2 fixed unassigned reshapes in em (#3674)

parent 09f9b35b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
modules/ml/src/em.cpp
View file @ 2bacd8b7
...@@ -135,7 +135,7 @@ Vec2d EM::predict(InputArray _sample, OutputArray _probs) const ...@@ -135,7 +135,7 @@ Vec2d EM::predict(InputArray _sample, OutputArray _probs) const
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() )
...@@ -266,7 +266,7 @@ void EM::setTrainData(int startStep, const Mat& samples, ...@@ -266,7 +266,7 @@ void EM::setTrainData(int startStep, const Mat& samples,
if(weights0 && (startStep == EM::START_E_STEP && covs0)) if(weights0 && (startStep == EM::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);
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment