typing for unproject_points

Summary: Fix the base class annotation for unproject_points.

Reviewed By: patricklabatut

Differential Revision: D33281586

fbshipit-source-id: 1c34e8c4b30b359fcb9307507bc778ad3fecf290
This commit is contained in:
Jeremy Reizenstein 2022-01-24 10:51:03 -08:00 committed by Facebook GitHub Bot
parent 67778caee8
commit 5053142363
2 changed files with 2 additions and 2 deletions

View File

@ -98,7 +98,7 @@ class CamerasBase(TensorProperties):
""" """
raise NotImplementedError() raise NotImplementedError()
def unproject_points(self): def unproject_points(self, xy_depth: torch.Tensor, **kwargs):
""" """
Transform input points from camera coodinates (NDC or screen) Transform input points from camera coodinates (NDC or screen)
to the world / camera coordinates. to the world / camera coordinates.

View File

@ -549,7 +549,7 @@ def _xy_to_ray_bundle(
) )
# unproject the points # unproject the points
unprojected = cameras.unproject_points(to_unproject, from_ndc=True) # pyre-ignore unprojected = cameras.unproject_points(to_unproject, from_ndc=True)
# split the two planes back # split the two planes back
rays_plane_1_world = unprojected[:, :n_rays_per_image] rays_plane_1_world = unprojected[:, :n_rays_per_image]