From cd78b782b0c0bfa7cf9e32bcbb276e4afbfb1df9 Mon Sep 17 00:00:00 2001 From: Ruilong Li Date: Sun, 9 Apr 2023 10:43:16 +0000 Subject: [PATCH] allows to fall back to the pytorch undistortion --- nerfstudio/cameras/cameras.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/nerfstudio/cameras/cameras.py b/nerfstudio/cameras/cameras.py index 8d3378fbc..3ab732155 100644 --- a/nerfstudio/cameras/cameras.py +++ b/nerfstudio/cameras/cameras.py @@ -654,16 +654,16 @@ class Cameras(TensorDataclass): ).reshape(-1, 2) else: # try to use nerfacc to accelerate if we don't need to optimize distortion params - # try: - coord_stack[coord_mask, :] = opencv_lens_undistortion( - coord_stack[coord_mask, :].reshape(3, -1, 2), - distortion_params[mask, :], - ).reshape(-1, 2) - # except: - # coord_stack[coord_mask, :] = camera_utils.radial_and_tangential_undistort( - # coord_stack[coord_mask, :].reshape(3, -1, 2), - # distortion_params[mask, :], - # ).reshape(-1, 2) + try: + coord_stack[coord_mask, :] = opencv_lens_undistortion( + coord_stack[coord_mask, :].reshape(3, -1, 2), + distortion_params[mask, :], + ).reshape(-1, 2) + except: + coord_stack[coord_mask, :] = camera_utils.radial_and_tangential_undistort( + coord_stack[coord_mask, :].reshape(3, -1, 2), + distortion_params[mask, :], + ).reshape(-1, 2) mask = (self.camera_type[true_indices] == CameraType.FISHEYE.value).squeeze(-1) # (num_rays) coord_mask = torch.stack([mask, mask, mask], dim=0)