Updating faces instead of vertices

eduardramon · Dec 24, 2020 · 3242df6 · 3242df6
1 parent 82b6de6
commit 3242df6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/code/evaluation/eval.py b/code/evaluation/eval.py
@@ -105,7 +105,7 @@ def evaluate(**kwargs):
             cams_transformation[:3,:3] = c_opt * R_opt
             cams_transformation[:3,3] = t_opt
 
-        mesh = plt.get_surface_high_res_mesh(
+        mesh = plt.get_surface_mesh(
             sdf=lambda x: model.geometry_network(x)[:, 0],
             resolution=kwargs['resolution']
         )

diff --git a/code/utils/plots.py b/code/utils/plots.py
@@ -176,8 +176,9 @@ def get_surface_mesh(sdf, resolution=100, sphere_radius=1.):
     mesh = components[areas.argmax()]
 
     if sphere_radius:
-        keep_vertices_idx = np.linalg.norm(mesh.vertices, axis=-1) < sphere_radius
-        mesh.update_vertices(keep_vertices_idx)
+        vertices_mask = np.linalg.norm(mesh.vertices, axis=-1) < sphere_radius
+        faces_mask = vertices_mask[mesh.faces].all(axis=1)
+        mesh.update_faces(faces_mask)
 
     return mesh