Update geom_solver.py

geom_solver.py

@@ -8,6 +8,18 @@ import torch
 from pytorch3d.renderer import PerspectiveCameras
 
 
+def my_empty_solution():
+	'''Return a minimal valid solution, i.e. 2 vertices and 1 edge.'''
+	# return np.zeros((1,3)), [(0, 0)]
+	return np.zeros((17,3)), [(0, 1)]
+
+
+def mean_solution(pcloud):
+	mean_point = pcloud.mean(axis=0)
+	verts = mean_point.repeat(0, 2)
+	return verts, [(0, 0)]
+
+
 class GeomSolver(object):
 
 	def __init__(self, entry):
@@ -110,11 +122,9 @@ class GeomSolver(object):
 	def get_vertices(self, visualize=False):
 		if visualize:
 			from hoho.viz3d import plot_estimate_and_gt
-			plot_estimate_and_gt(self.vertices, [(0,1)], self.human_entry['wf_vertices'], self.human_entry['wf_edges'])
+			plot_estimate_and_gt(self.vertices, [(0,0)], self.human_entry['wf_vertices'], self.human_entry['wf_edges'])
 		if self.vertices.shape[0] == 0:
 			return my_empty_solution()
-		return self.vertices, [(0, len(self.vertices)-1)]
-
-
+		return self.vertices, [(0, 0)]