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

Daily arc lint --take BLACK

Browse Source 
Reviewed By: wynsmart

Differential Revision: D30065248

fbshipit-source-id: 600915ab43d3d6d4846f60f976391f9dc1d77d10
This commit is contained in:
CodemodService Bot 2021-08-03 04:33:32 -07:00 committed by Facebook GitHub Bot
parent 0c32f094af
commit 55aaec4d83
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pytorch3d/renderer/mesh/rasterizer.py
View File

@ -127,7 +127,9 @@ class MeshRasterizer(nn.Module):
) )
# view to NDC transform # view to NDC transform
to_ndc_transform = cameras.get_ndc_camera_transform(**kwargs) to_ndc_transform = cameras.get_ndc_camera_transform(**kwargs)
projection_transform = cameras.get_projection_transform(**kwargs).compose(to_ndc_transform) projection_transform = cameras.get_projection_transform(**kwargs).compose(
to_ndc_transform
)
verts_ndc = projection_transform.transform_points(verts_view, eps=eps) verts_ndc = projection_transform.transform_points(verts_view, eps=eps)
verts_ndc[..., 2] = verts_view[..., 2] verts_ndc[..., 2] = verts_view[..., 2]

4
pytorch3d/renderer/points/rasterizer.py
View File

@ -98,7 +98,9 @@ class PointsRasterizer(nn.Module):
) )
# view to NDC transform # view to NDC transform
to_ndc_transform = cameras.get_ndc_camera_transform(**kwargs) to_ndc_transform = cameras.get_ndc_camera_transform(**kwargs)
projection_transform = cameras.get_projection_transform(**kwargs).compose(to_ndc_transform) projection_transform = cameras.get_projection_transform(**kwargs).compose(
to_ndc_transform
)
pts_ndc = projection_transform.transform_points(pts_view, eps=eps) pts_ndc = projection_transform.transform_points(pts_view, eps=eps)
pts_ndc[..., 2] = pts_view[..., 2] pts_ndc[..., 2] = pts_view[..., 2]