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
This commit is contained in:
Jeremy Reizenstein 2023-11-14 07:42:18 -08:00 committed by Facebook GitHub Bot
parent f613682551
commit f4f2209271
2 changed files with 2 additions and 3 deletions

View File

@ -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

View File

@ -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:
"""