Skip to content

Commit f4f2209

Browse files
bottlerfacebook-github-bot
authored andcommitted
Fix for mask_points=False
Summary: Remove unused argument `mask_points` from `get_rgbd_point_cloud` and fix `get_implicitron_sequence_pointcloud`, which assumed it was used. Reviewed By: MichaelRamamonjisoa Differential Revision: D50885848 fbshipit-source-id: c0b834764ad5ef560107bd8eab04952d000489b8
1 parent f613682 commit f4f2209

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

pytorch3d/implicitron/dataset/visualize.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -89,9 +89,8 @@ def get_implicitron_sequence_pointcloud(
8989
frame_data.image_rgb,
9090
frame_data.depth_map,
9191
(cast(torch.Tensor, frame_data.fg_probability) > 0.5).float()
92-
if frame_data.fg_probability is not None
92+
if mask_points and frame_data.fg_probability is not None
9393
else None,
94-
mask_points=mask_points,
9594
)
9695

9796
return point_cloud, frame_data

pytorch3d/implicitron/tools/point_cloud_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def get_rgbd_point_cloud(
2626
depth_map: torch.Tensor,
2727
mask: Optional[torch.Tensor] = None,
2828
mask_thr: float = 0.5,
29-
mask_points: bool = True,
29+
*,
3030
euclidean: bool = False,
3131
) -> Pointclouds:
3232
"""

0 commit comments

Comments
 (0)