upgrade pyre version in fbcode/vision - batch 2

Reviewed By: bottler

Differential Revision: D60992234

fbshipit-source-id: 899db6ed590ef966ff651c11027819e59b8401a3
This commit is contained in:
generatedunixname89002005307016
2024-08-09 02:07:45 -07:00
committed by Facebook GitHub Bot
parent 1e0b1d9c72
commit 38afdcfc68
26 changed files with 87 additions and 27 deletions

View File

@@ -156,7 +156,6 @@ def render_point_cloud_pytorch3d(
cumprod = torch.cat((torch.ones_like(cumprod[..., :1]), cumprod[..., :-1]), dim=-1)
depths = (weights * cumprod * fragments.zbuf).sum(dim=-1)
# add the rendering mask
# pyre-fixme[6]: For 1st param expected `Tensor` but got `float`.
render_mask = -torch.prod(1.0 - weights, dim=-1) + 1.0
# cat depths and render mask