mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-13 11:24:00 +08:00
Fix ruff warnings
This commit is contained in:
1 parent
09561e4272
commit
f70d3946bc
1 file changed
+2
-3
@@ -24,7 +24,6 @@ import math
|
||||
import numpy as np
|
||||
|
||||
from nerfstudio.process_data.process_data_utils import CAMERA_MODELS
|
||||
from nerfstudio.utils.rich_utils import CONSOLE
|
||||
|
||||
def rodrigues_vec_to_rotation_mat(rodrigues_vec):
|
||||
theta = np.linalg.norm(rodrigues_vec)
|
||||
@@ -32,7 +31,7 @@ def rodrigues_vec_to_rotation_mat(rodrigues_vec):
|
||||
rotation_mat = np.eye(3, dtype=float)
|
||||
else:
|
||||
r = rodrigues_vec / theta
|
||||
I = np.eye(3, dtype=float)
|
||||
ident = np.eye(3, dtype=float)
|
||||
r_rT = np.array([
|
||||
[r[0]*r[0], r[0]*r[1], r[0]*r[2]],
|
||||
[r[1]*r[0], r[1]*r[1], r[1]*r[2]],
|
||||
@@ -43,7 +42,7 @@ def rodrigues_vec_to_rotation_mat(rodrigues_vec):
|
||||
[r[2], 0, -r[0]],
|
||||
[-r[1], r[0], 0]
|
||||
])
|
||||
rotation_mat = math.cos(theta) * I + (1 - math.cos(theta)) * r_rT + math.sin(theta) * r_cross
|
||||
rotation_mat = math.cos(theta) * ident + (1 - math.cos(theta)) * r_rT + math.sin(theta) * r_cross
|
||||
return rotation_mat
|
||||
|
||||
def cameras2nerfds(
|
||||
|
||||
Reference in new issue
Block a user