mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 20:02:49 +08:00
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:
parent
67778caee8
commit
5053142363
@ -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.
|
||||||
|
@ -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]
|
||||||
|
Loading…
x
Reference in New Issue
Block a user