Upload pytorch3d_stub/pytorch3d/structures/__init__.py with huggingface_hub

pytorch3d_stub/pytorch3d/structures/__init__.py

@@ -0,0 +1,40 @@
+"""pytorch3d.structures stub – Meshes only."""
+import torch
+
+class Meshes:
+    def __init__(self, verts=None, faces=None, textures=None):
+        if isinstance(verts, list):
+            self._verts_list = [v if isinstance(v, torch.Tensor) else torch.tensor(v, dtype=torch.float32) for v in verts]
+        elif isinstance(verts, torch.Tensor):
+            self._verts_list = [verts]
+        else:
+            self._verts_list = []
+        if isinstance(faces, list):
+            self._faces_list = [f if isinstance(f, torch.Tensor) else torch.tensor(f, dtype=torch.long) for f in faces]
+        elif isinstance(faces, torch.Tensor):
+            self._faces_list = [faces]
+        else:
+            self._faces_list = []
+        self.textures = textures
+
+    def verts_list(self):
+        return self._verts_list
+
+    def faces_list(self):
+        return self._faces_list
+
+    def verts_packed(self):
+        if self._verts_list:
+            return torch.cat(self._verts_list, dim=0)
+        return torch.zeros(0, 3)
+
+    def faces_packed(self):
+        if self._faces_list:
+            return torch.cat(self._faces_list, dim=0)
+        return torch.zeros(0, 3, dtype=torch.long)
+
+    def num_verts_per_mesh(self):
+        return torch.tensor([v.shape[0] for v in self._verts_list])
+
+    def __len__(self):
+        return len(self._verts_list)