update

geom_solver.py

@@ -16,7 +16,7 @@ def my_empty_solution():
 class GeomSolver(object):
 
 	def __init__(self):
-		self.min_vertices = 2
+		self.min_vertices = 18
 		self.kmeans_th = 130
 		self.point_dist_th = 60
 		self.th_min_support = 3
@@ -147,13 +147,18 @@ class GeomSolver(object):
 			vertices = centers
 			print("Enough vertices.")
 		else:
-			# if self.repeat_predicted:
-			if self.mean_fixed:
-				added_one = (self.min_vertices * self.wf_center - self.vertices.sum(0)) / (self.min_vertices - nvert)
+			vertices = centers
+			if self.repeat_predicted:
+				while vertices.shape[0] < self.min_vertices:
+					vertices = np.concatenate((vertices, centers))
+				vertices = vertices[:self.min_vertices]
 			else:
-				added_one = self.wf_center
-			added = added_one[None].repeat(self.min_vertices - nvert,0)
-			vertices = np.concatenate((self.vertices, added))
+				if self.mean_fixed:
+					added_one = (self.min_vertices * self.wf_center - self.vertices.sum(0)) / (self.min_vertices - nvert)
+				else:
+					added_one = self.wf_center
+				added = added_one[None].repeat(self.min_vertices - nvert,0)
+				vertices = np.concatenate((self.vertices, added))
 		self.vertices_aug = vertices