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

almost finished Python wrappers

Browse Source
This commit is contained in:
Vadim Pisarevsky
2010-11-02 17:58:22 +00:00
parent eda72a3e8b
commit 5b6a755719
27 changed files with 1292 additions and 634 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1109,14 +1109,13 @@ const CvMat* CvEM::get_probs() const
using namespace cv;
CvEM::CvEM( const Mat& samples, const Mat& sample_idx,
CvEMParams params, Mat* labels )
CvEM::CvEM( const Mat& samples, const Mat& sample_idx, CvEMParams params )
{
means = weights = probs = inv_eigen_values = log_weight_div_det = 0;
covs = cov_rotate_mats = 0;
// just invoke the train() method
train(samples, sample_idx, params, labels);
train(samples, sample_idx, params);
}
bool CvEM::train( const Mat& _samples, const Mat& _sample_idx,
@@ -1154,5 +1153,33 @@ CvEM::predict( const Mat& _sample, Mat* _probs ) const
return predict(&sample, pprobs);
}
int CvEM::getNClusters() const
{
return params.nclusters;
}
Mat CvEM::getMeans() const
{
return Mat(means);
}
void CvEM::getCovs(vector<Mat>& _covs) const
{
int i, n = params.nclusters;
_covs.resize(n);
for( i = 0; i < n; i++ )
_covs[i] = Mat(covs[i]);
}
Mat CvEM::getWeights() const
{
return Mat(weights);
}
Mat CvEM::getProbs() const
{
return Mat(probs);
}
/* End of file. */