From 080eef4d787a02a1a746384b480646d69303ade7 Mon Sep 17 00:00:00 2001 From: Piero Toffanin Date: Thu, 12 Oct 2023 17:59:18 -0400 Subject: [PATCH] Add annotations --- nerfstudio/process_data/odm_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/process_data/odm_utils.py b/nerfstudio/process_data/odm_utils.py index 7d1fd184c..96d0448aa 100644 --- a/nerfstudio/process_data/odm_utils.py +++ b/nerfstudio/process_data/odm_utils.py @@ -26,7 +26,7 @@ import numpy as np from nerfstudio.process_data.process_data_utils import CAMERA_MODELS -def rodrigues_vec_to_rotation_mat(rodrigues_vec): +def rodrigues_vec_to_rotation_mat(rodrigues_vec: np.ndarray) -> np.ndarray: theta = np.linalg.norm(rodrigues_vec) if theta < sys.float_info.epsilon: rotation_mat = np.eye(3, dtype=float)