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

typing for unproject_points

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch3d/renderer/cameras.py
View File

@ -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.

2
pytorch3d/renderer/implicit/raysampling.py
View File

@ -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]