update

asset3d_gen/data/mesh_operator.py

@@ -68,28 +68,18 @@ class MeshFixer(object):
 
     def __init__(
         self,
+        vertices: Union[torch.Tensor, np.ndarray],
+        faces: Union[torch.Tensor, np.ndarray],
         device: str = "cuda",
     ) -> None:
         self.device = device
-        self.vertices = None
-        self.faces = None
-    
-    @spaces.GPU
-    def set_mesh(
-        self,
-        vertices: Union[torch.Tensor, np.ndarray],
-        faces: Union[torch.Tensor, np.ndarray],
-    ):
-        self.vertices = (
-            torch.tensor(vertices, device=self.device)
-            if isinstance(vertices, np.ndarray)
-            else vertices.to(self.device)
-        )
-        self.faces = (
-            torch.tensor(faces.astype(np.int32), device=self.device)
-            if isinstance(faces, np.ndarray)
-            else faces.to(self.device)
-        )
+        if isinstance(vertices, np.ndarray):
+            vertices = torch.tensor(vertices)
+        self.vertices = vertices
+
+        if isinstance(faces, np.ndarray):
+            faces = torch.tensor(faces)
+        self.faces = faces
 
     @staticmethod
     def log_mesh_changes(method):
@@ -421,6 +411,8 @@ class MeshFixer(object):
                 - vertices: Simplified and repaired vertex array of (V, 3).
                 - faces: Simplified and repaired face array of (F, 3).
         """
+        self.vertices.to(self.device)
+        self.faces.to(self.device)
         self.simplify(ratio=filter_ratio)
         self.fill_holes(
             max_hole_size=max_hole_size,