fit spherical GMM so it runs faster

Former-commit-id: a6db2b891f7f2f89aab453bcb4a736de14dbe43c
javiribera · Oct 8, 2018 · d5b6c0d · d5b6c0d
1 parent e402b15
commit d5b6c0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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