diff --git a/pytorch3d/renderer/mesh/rasterizer.py b/pytorch3d/renderer/mesh/rasterizer.py index d016ff27..9e49124a 100644 --- a/pytorch3d/renderer/mesh/rasterizer.py +++ b/pytorch3d/renderer/mesh/rasterizer.py @@ -127,7 +127,9 @@ class MeshRasterizer(nn.Module): ) # view to NDC transform 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[..., 2] = verts_view[..., 2] diff --git a/pytorch3d/renderer/points/rasterizer.py b/pytorch3d/renderer/points/rasterizer.py index 55e1e876..fe3dfd51 100644 --- a/pytorch3d/renderer/points/rasterizer.py +++ b/pytorch3d/renderer/points/rasterizer.py @@ -98,7 +98,9 @@ class PointsRasterizer(nn.Module): ) # view to NDC transform 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[..., 2] = pts_view[..., 2]