Update geom_solver.py
Browse files- geom_solver.py +2 -2
geom_solver.py
CHANGED
@@ -18,10 +18,10 @@ class GeomSolver(object):
|
|
18 |
def __init__(self):
|
19 |
self.min_vertices = 18
|
20 |
self.kmeans_th = 70
|
21 |
-
|
22 |
self.clr_th = 2.5
|
23 |
self.device = 'cuda:0'
|
24 |
-
|
25 |
|
26 |
def cluster_points(self, point_types):
|
27 |
point_colors = []
|
|
|
18 |
def __init__(self):
|
19 |
self.min_vertices = 18
|
20 |
self.kmeans_th = 70
|
21 |
+
self.point_dist_th = 15
|
22 |
self.clr_th = 2.5
|
23 |
self.device = 'cuda:0'
|
24 |
+
self.return_edges = False
|
25 |
|
26 |
def cluster_points(self, point_types):
|
27 |
point_colors = []
|