Update geom_solver.py
Browse files- geom_solver.py +5 -5
geom_solver.py
CHANGED
@@ -222,11 +222,11 @@ class GeomSolver(object):
|
|
222 |
self.human_entry = human_entry
|
223 |
self.process_vertices()
|
224 |
vertices = self.vertices_aug
|
225 |
-
|
226 |
-
|
227 |
-
|
228 |
-
|
229 |
-
|
230 |
if visualize:
|
231 |
from hoho.viz3d import plot_estimate_and_gt
|
232 |
plot_estimate_and_gt(vertices, [(0,0)], self.human_entry['wf_vertices'], self.human_entry['wf_edges'])
|
|
|
222 |
self.human_entry = human_entry
|
223 |
self.process_vertices()
|
224 |
vertices = self.vertices_aug
|
225 |
+
if self.return_edges:
|
226 |
+
edges = self.process_edges()
|
227 |
+
else:
|
228 |
+
edges = [(0, 0)]
|
229 |
+
|
230 |
if visualize:
|
231 |
from hoho.viz3d import plot_estimate_and_gt
|
232 |
plot_estimate_and_gt(vertices, [(0,0)], self.human_entry['wf_vertices'], self.human_entry['wf_edges'])
|