Update pointcloudfunc.py

Signed-off-by: nirmal <[email protected]>
NVIDIAGameWorks · May 8, 2020 · a61f4eb · a61f4eb
1 parent c3ec548
commit a61f4eb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/kaolin/transforms/pointcloudfunc.py b/kaolin/transforms/pointcloudfunc.py
@@ -196,6 +196,8 @@ def normalize(cloud: Union[torch.Tensor, PointCloud],
 
     if isinstance(cloud, np.ndarray):
         cloud = torch.from_numpy(cloud)
+    if isinstance(cloud, PointCloud):
+        cloud = cloud.points
 
     helpers._assert_tensor(cloud)
     helpers._assert_dim_ge(cloud, 2)