From: Maria Dimashova Date: Wed, 18 Apr 2012 11:56:50 +0000 (+0000) Subject: made em params read only; minor renaming X-Git-Tag: accepted/tizen/6.0/unified/20201030.111113~1314^2~2054 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=ca8241f6e58b807f2e46fdbf3699546cc6c9709c;p=platform%2Fupstream%2Fopencv.git made em params read only; minor renaming --- diff --git a/modules/ml/include/opencv2/ml/ml.hpp b/modules/ml/include/opencv2/ml/ml.hpp index da83cf4..0cc1b5f 100644 --- a/modules/ml/include/opencv2/ml/ml.hpp +++ b/modules/ml/include/opencv2/ml/ml.hpp @@ -569,7 +569,7 @@ public: enum {START_E_STEP=1, START_M_STEP=2, START_AUTO_STEP=0}; CV_WRAP EM(int nclusters=EM::DEFAULT_NCLUSTERS, int covMatType=EM::COV_MAT_DIAGONAL, - const TermCriteria& termcrit=TermCriteria(TermCriteria::COUNT+ + const TermCriteria& termCrit=TermCriteria(TermCriteria::COUNT+ TermCriteria::EPS, EM::DEFAULT_MAX_ITERS, FLT_EPSILON)); diff --git a/modules/ml/src/em.cpp b/modules/ml/src/em.cpp index 7583ea1..38adae2 100644 --- a/modules/ml/src/em.cpp +++ b/modules/ml/src/em.cpp @@ -48,12 +48,12 @@ const double minEigenValue = DBL_EPSILON; /////////////////////////////////////////////////////////////////////////////////////////////////////// -EM::EM(int _nclusters, int _covMatType, const TermCriteria& _criteria) +EM::EM(int _nclusters, int _covMatType, const TermCriteria& _termCrit) { nclusters = _nclusters; covMatType = _covMatType; - maxIters = (_criteria.type & TermCriteria::MAX_ITER) ? _criteria.maxCount : DEFAULT_MAX_ITERS; - epsilon = (_criteria.type & TermCriteria::EPS) ? _criteria.epsilon : 0; + maxIters = (_termCrit.type & TermCriteria::MAX_ITER) ? _termCrit.maxCount : DEFAULT_MAX_ITERS; + epsilon = (_termCrit.type & TermCriteria::EPS) ? _termCrit.epsilon : 0; } EM::~EM() @@ -135,6 +135,7 @@ Vec2d EM::predict(InputArray _sample, OutputArray _probs) const sample.convertTo(tmp, CV_64FC1); sample = tmp; } + sample.reshape(1, 1); Mat probs; if( _probs.needed() ) @@ -682,12 +683,12 @@ AlgorithmInfo* EM::info() const if( !initialized ) { EM obj; - em_info.addParam(obj, "nclusters", obj.nclusters); - em_info.addParam(obj, "covMatType", obj.covMatType); + em_info.addParam(obj, "nclusters", obj.nclusters, true); + em_info.addParam(obj, "covMatType", obj.covMatType, true); - em_info.addParam(obj, "weights", obj.weights); - em_info.addParam(obj, "means", obj.means); - em_info.addParam(obj, "covs", obj.covs); + em_info.addParam(obj, "weights", obj.weights, true); + em_info.addParam(obj, "means", obj.means, true); + em_info.addParam(obj, "covs", obj.covs, true); initialized = true; }