diff --git a/nerfstudio/process_data/odm_utils.py b/nerfstudio/process_data/odm_utils.py index aa9b69541..57294776f 100644 --- a/nerfstudio/process_data/odm_utils.py +++ b/nerfstudio/process_data/odm_utils.py @@ -38,7 +38,8 @@ def rodrigues_vec_to_rotation_mat(rodrigues_vec: np.ndarray) -> np.ndarray: [r[0] * r[0], r[0] * r[1], r[0] * r[2]], [r[1] * r[0], r[1] * r[1], r[1] * r[2]], [r[2] * r[0], r[2] * r[1], r[2] * r[2]], - ] + ], + dtype=float ) r_cross = np.array([[0, -r[2], r[1]], [r[2], 0, -r[0]], [-r[1], r[0], 0]], dtype=float) rotation_mat = math.cos(theta) * ident + (1 - math.cos(theta)) * r_rT + math.sin(theta) * r_cross