Adapt to ZeroGPU (removed open3d)?

utils/mesh.py

@@ -3,11 +3,11 @@ from pathlib import Path
 import os 
 from tqdm import tqdm
 import potpourri3d as pp3d
-import open3d as o3d
 import scipy.io as sio
 import numpy as np
 import re
 import sys
+from utils.surfaces import Surface, centroid
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 from shape_data import get_data_dirs
 
@@ -102,37 +102,21 @@ def save_ply(file_name, V, F, Rho=None, color=None):
     file.writelines(lines)
     file.close()
 
-def numpy_to_open3d_mesh(V, F):
-    # Create an empty TriangleMesh object
-    mesh = o3d.geometry.TriangleMesh()
-    # Set vertices
-    mesh.vertices = o3d.utility.Vector3dVector(V)
-    # Set triangles
-    mesh.triangles = o3d.utility.Vector3iVector(F)
-    return mesh
-
 
 
 def load_mesh(filepath, scale=True, return_vnormals=False):
     V, F = pp3d.read_mesh(filepath)
-    mesh = numpy_to_open3d_mesh(V, F)
-
-    tmat = np.identity(4, dtype=np.float32)
-    center = mesh.get_center()
-    tmat[:3, 3] = -center
-    area = mesh.get_surface_area()
+    surf = Surface(FV=[F,V])
+    center, area = centroid(surf)
     if scale:
-        smat = np.identity(4, dtype=np.float32)
-        
-        smat[:3, :3] = np.identity(3, dtype=np.float32) / np.sqrt(area)
-        tmat = smat @ tmat
-    mesh.transform(tmat)
-
-    vertices = np.asarray(mesh.vertices, dtype=np.float32)
-    faces = np.asarray(mesh.triangles, dtype=np.int32)
+        new_vertices = (surf.vertices - center)/area
+    else:
+        new_vertices = (surf.vertices - center)
+    vertices = np.asarray(new_vertices, dtype=np.float32)
+    faces = np.asarray(F, dtype=np.int32)
+
     if return_vnormals:
-        mesh.compute_vertex_normals()
-        vnormals = np.asarray(mesh.vertex_normals, dtype=np.float32)
+        vnormals = surf.computeVertexNormals()
         if scale:
             return vertices, faces, vnormals, area, center
         return vertices, faces, vnormals, area, center