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)