cameras_from_opencv_projection device #1021

Summary: Fix https://github.com/facebookresearch/pytorch3d/issues/1021 that cameras_from_opencv_projection always creates on CPU.

Reviewed By: nikhilaravi

Differential Revision: D33508211

fbshipit-source-id: fadebd45cacafd633af6a58094cf6f654529992c
This commit is contained in:
Jeremy Reizenstein
2022-01-21 05:29:04 -08:00
committed by Facebook GitHub Bot
parent 39bb2ce063
commit 45d096e219
2 changed files with 8 additions and 4 deletions

View File

@@ -55,6 +55,7 @@ def _cameras_from_opencv_projection(
focal_length=focal_pytorch3d,
principal_point=p0_pytorch3d,
image_size=image_size,
device=R.device,
)