From d5b6c0d0db346373ac20cbf75468784877ce1f15 Mon Sep 17 00:00:00 2001 From: Javi Ribera Date: Sun, 7 Oct 2018 20:20:20 -0400 Subject: [PATCH] fit spherical GMM so it runs faster Former-commit-id: a6db2b891f7f2f89aab453bcb4a736de14dbe43c --- object-locator/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/object-locator/utils.py b/object-locator/utils.py index 14053a7..16cccea 100644 --- a/object-locator/utils.py +++ b/object-locator/utils.py @@ -109,7 +109,7 @@ def cluster(array, n_clusters, max_mask_pts=np.infty): n_components = max(min(n_clusters, x.size), 1) centroids = sklearn.mixture.GaussianMixture(n_components=n_components, n_init=1, - covariance_type='full').\ + covariance_type='spherical').\ fit(c).means_.astype(np.int) return centroids