diff --git a/pytorch3d/implicitron/dataset/visualize.py b/pytorch3d/implicitron/dataset/visualize.py index 8c277532..4ac633f6 100644 --- a/pytorch3d/implicitron/dataset/visualize.py +++ b/pytorch3d/implicitron/dataset/visualize.py @@ -89,9 +89,8 @@ def get_implicitron_sequence_pointcloud( frame_data.image_rgb, frame_data.depth_map, (cast(torch.Tensor, frame_data.fg_probability) > 0.5).float() - if frame_data.fg_probability is not None + if mask_points and frame_data.fg_probability is not None else None, - mask_points=mask_points, ) return point_cloud, frame_data diff --git a/pytorch3d/implicitron/tools/point_cloud_utils.py b/pytorch3d/implicitron/tools/point_cloud_utils.py index af2abe64..904e4567 100644 --- a/pytorch3d/implicitron/tools/point_cloud_utils.py +++ b/pytorch3d/implicitron/tools/point_cloud_utils.py @@ -26,7 +26,7 @@ def get_rgbd_point_cloud( depth_map: torch.Tensor, mask: Optional[torch.Tensor] = None, mask_thr: float = 0.5, - mask_points: bool = True, + *, euclidean: bool = False, ) -> Pointclouds: """