423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-22 15:20:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Alpha channel to return the mask

Browse Source 
Summary: Updated the alpha channel in the `hard_rgb_blend` function to return the mask of the pixels which have overlapping mesh faces.

Reviewed By: bottler

Differential Revision: D29001604

fbshipit-source-id: 22a2173d769f2d3ad34892d68ceb628f073bca22
This commit is contained in:
Nikhila Ravi
2021-06-09 15:06:03 -07:00
committed by Facebook GitHub Bot
parent ac6c07fa43
commit a15c33a3cc
3 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pytorch3d/renderer/blending.py
View File

@@ -58,7 +58,8 @@ def hard_rgb_blend(colors, fragments, blend_params) -> torch.Tensor:
) # (N, H, W, 3)
# Concat with the alpha channel.
alpha = torch.ones((N, H, W, 1), dtype=colors.dtype, device=device)
alpha = (~is_background).type_as(pixel_colors)[..., None]
return torch.cat([pixel_colors, alpha], dim=-1) # (N, H, W, 4)