jboth commited on
Commit
7a4ca28
·
verified ·
1 Parent(s): c580568

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

Browse files
pytorch3d_stub/pytorch3d/structures/__init__.py CHANGED
@@ -38,3 +38,12 @@ class Meshes:
38
 
39
  def __len__(self):
40
  return len(self._verts_list)
 
 
 
 
 
 
 
 
 
 
38
 
39
  def __len__(self):
40
  return len(self._verts_list)
41
+
42
+ class Pointclouds:
43
+ """Minimal Pointclouds stub."""
44
+ def __init__(self, points=None, features=None, normals=None):
45
+ self.points_list = points if isinstance(points, list) else ([points] if points is not None else [])
46
+ self.features_list = features if isinstance(features, list) else ([features] if features is not None else [])
47
+ self.normals_list = normals if isinstance(normals, list) else ([normals] if normals is not None else [])
48
+ def to(self, device):
49
+ return self