diff --git a/pytorch3d/renderer/cameras.py b/pytorch3d/renderer/cameras.py index 1225381e..a840d54b 100644 --- a/pytorch3d/renderer/cameras.py +++ b/pytorch3d/renderer/cameras.py @@ -98,7 +98,7 @@ class CamerasBase(TensorProperties): """ 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 c4e4139d..e868a18d 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]