feat: automatic conversion to numpy

seung-lab · Nov 7, 2024 · 815fe6c · 815fe6c
1 parent c0d2ac2
commit 815fe6c
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/zmesh/mesh.py b/zmesh/mesh.py
@@ -14,9 +14,14 @@ class Mesh:
 
   """
   def __init__(self, vertices, faces, normals, id=None):
-    self.vertices = vertices
-    self.faces = faces
-    self.normals = normals
+    self.vertices = np.asarray(vertices, dtype=np.float32)
+    self.faces = np.asarray(faces, dtype=np.uint32)
+
+    if normals is None:
+      self.normals = normals
+    else:
+      self.normals = np.asarray(normals, dtype=np.float32)
+
     self.id = id
 
   def __len__(self):