mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-01 03:12:49 +08:00
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:
parent
f613682551
commit
f4f2209271
@ -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:
|
||||
"""
|
||||
|
Loading…
x
Reference in New Issue
Block a user