423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 03:42:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix for mask_points=False

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pytorch3d/implicitron/dataset/visualize.py
View File

@ -89,9 +89,8 @@ def get_implicitron_sequence_pointcloud(
frame_data.image_rgb, frame_data.image_rgb,
frame_data.depth_map, frame_data.depth_map,
(cast(torch.Tensor, frame_data.fg_probability) > 0.5).float() (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, else None,
mask_points=mask_points,
) )
return point_cloud, frame_data return point_cloud, frame_data

2
pytorch3d/implicitron/tools/point_cloud_utils.py
View File

@ -26,7 +26,7 @@ def get_rgbd_point_cloud(
depth_map: torch.Tensor, depth_map: torch.Tensor,
mask: Optional[torch.Tensor] = None, mask: Optional[torch.Tensor] = None,
mask_thr: float = 0.5, mask_thr: float = 0.5,
mask_points: bool = True, *,
euclidean: bool = False, euclidean: bool = False,
) -> Pointclouds: ) -> Pointclouds:
""" """