diff --git a/pytorch3d/renderer/cameras.py b/pytorch3d/renderer/cameras.py index 1225381ee..a840d54b4 100644 --- a/pytorch3d/renderer/cameras.py +++ b/pytorch3d/renderer/cameras.py @@ -98,7 +98,7 @@ def get_projection_transform(self): """ raise NotImplementedError() - def unproject_points(self): + def unproject_points(self, xy_depth: torch.Tensor, **kwargs): """ Transform input points from camera coodinates (NDC or screen) to the world / camera coordinates. diff --git a/pytorch3d/renderer/implicit/raysampling.py b/pytorch3d/renderer/implicit/raysampling.py index c4e4139d2..e868a18d3 100644 --- a/pytorch3d/renderer/implicit/raysampling.py +++ b/pytorch3d/renderer/implicit/raysampling.py @@ -549,7 +549,7 @@ def _xy_to_ray_bundle( ) # 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 rays_plane_1_world = unprojected[:, :n_rays_per_image]