update

geom_solver.py

@@ -16,7 +16,7 @@ def my_empty_solution():
 class GeomSolver(object):
 
 	def __init__(self):
-		self.min_vertices = 18
+		self.min_vertices = 2
 		self.kmeans_th = 130
 		self.point_dist_th = 60
 		self.th_min_support = 3
@@ -24,6 +24,7 @@ class GeomSolver(object):
 		self.device = 'cuda:0'
 		self.return_edges = False
 		self.mean_fixed = False
+		self.repeat_predicted = True
 
 	def cluster_points(self, point_types):
 		point_colors = []
@@ -138,11 +139,15 @@ class GeomSolver(object):
 		self.wf_center[-1] = centers[:, -1].mean()
 
 		self.vertices = centers
+		nvert = centers.shape[0]
 		if self.broken_cams.any():
 			vertices = centers
 			print("There are broken cams.")
+		elif nvert >= self.min_vertices:
+			vertices = centers
+			print("Enough vertices.")
 		else:
-			nvert = centers.shape[0]
+			# if self.repeat_predicted:
 			if self.mean_fixed:
 				added_one = (self.min_vertices * self.wf_center - self.vertices.sum(0)) / (self.min_vertices - nvert)
 			else: