Commit f4f22092 authored by Jeremy Reizenstein's avatar Jeremy Reizenstein Committed by Facebook GitHub Bot
Browse files

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
parent f6136825
......@@ -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
......@@ -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:
"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment