Update lib/dataset/mesh_util.py

lib/dataset/mesh_util.py

@@ -138,16 +138,14 @@ def mesh_edge_loss(meshes, target_length: float = 0.0):
 
 
 def remesh(obj_path, perc, device):
-
-    ms = pymeshlab.MeshSet()
-    ms.load_new_mesh(obj_path)
-    ms.laplacian_smooth()
-    ms.remeshing_isotropic_explicit_remeshing(
-        targetlen=pymeshlab.Percentage(perc), adaptive=True)
-    ms.save_current_mesh(obj_path.replace("recon", "remesh"))
-    polished_mesh = trimesh.load_mesh(obj_path.replace("recon", "remesh"))
-    verts_pr = torch.tensor(polished_mesh.vertices).float().unsqueeze(0).to(device)
-    faces_pr = torch.tensor(polished_mesh.faces).long().unsqueeze(0).to(device)
+    mesh = trimesh.load(obj_path)
+    mesh = mesh.simplify_quadratic_decimation(face_count)
+    mesh = trimesh.smoothing.filter_humphrey(
+        mesh, alpha=0.1, beta=0.5, iterations=10, laplacian_operator=None
+    )
+    mesh.export(obj_path.replace("recon", "remesh"))
+    verts_pr = torch.tensor(mesh.vertices).float().unsqueeze(0).to(device)
+    faces_pr = torch.tensor(mesh.faces).long().unsqueeze(0).to(device)
 
     return verts_pr, faces_pr