fixed gaussian_mix.py sample

13 years ago · ddf7fde3b2
parent 4793ca8fbb
commit ddf7fde3b2
1 changed files with 4 additions and 3 deletions
--- a/samples/python2/gaussian_mix.py
+++ b/samples/python2/gaussian_mix.py
@ -35,9 +35,10 @@ if __name__ == '__main__':
        points, ref_distrs = make_gaussians(cluster_n, img_size)

        print 'EM (opencv) ...'
-        em = cv2.EM(points, params = dict( nclusters = cluster_n, cov_mat_type = cv2.EM_COV_MAT_GENERIC) )
-        means = em.getMeans()
-        covs = em.getCovs()
+        em = cv2.EM(cluster_n, cv2.EM_COV_MAT_GENERIC)
+        em.train(points)
+        means = em.getMat('means')
+        covs = em.getMatVector('covs')
        found_distrs = zip(means, covs)
        print 'ready!\n'