Update geom_solver.py

geom_solver.py

@@ -143,10 +143,11 @@ class GeomSolver(object):
 		self.vertices = centers
 		nvert = centers.shape[0]
 		# desired_vertices = (self.xyz[:,-1] > z_th).sum() // 300
-		# if desired_vertices < self.min_vertices:
-		# 	desired_vertices = self.mean_vertices
-		# if desired_vertices > self.max_vertices:
-		desired_vertices = self.mean_vertices
+        desired_vertices = 2*nvert
+		if desired_vertices < self.min_vertices:
+			desired_vertices = self.mean_vertices
+		if desired_vertices > self.max_vertices:
+    		desired_vertices = self.mean_vertices
 		if self.broken_cams.any():
 			vertices = centers
 			print("There are broken cams.")
@@ -157,7 +158,7 @@ class GeomSolver(object):
 			vertices = centers
 			if self.repeat_predicted:
 				while vertices.shape[0] < desired_vertices:
-					vertices = np.concatenate((vertices, centers[~self.is_apex]))
+					vertices = np.concatenate((vertices, centers)) # [~self.is_apex]
 				vertices = vertices[:desired_vertices]
 			else:
 				if self.mean_fixed: