Skip to content

Commit b95d5b4

Browse files
committed
modify-calibration: Fix set_first_to_second mode JSON reading.
1 parent 0787733 commit b95d5b4

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

modify-calibration.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,16 +69,22 @@ def set_first_to_second_matrix(calibration, first_to_second, second_camera=False
6969
try:
7070
with open(args.matrix) as f:
7171
calibration = json.load(f)
72+
if args.action == 'set_imu_to_camera':
7273
matrix = calibration["cameras"][args.matrixIndexInCalibration]["imuToCamera"]
74+
else:
75+
assert(args.action == 'set_first_to_second')
76+
imuToCam0 = np.array(calibration["cameras"][0]["imuToCamera"])
77+
imuToCam1 = np.array(calibration["cameras"][1]["imuToCamera"])
78+
matrix = imuToCam1 @ np.linalg.inv(imuToCam0)
7379
except:
7480
matrix = json.loads(args.matrix)
7581

7682
calib_in = json.load(sys.stdin)
7783

7884
if args.action == 'set_imu_to_camera':
7985
calib_out = set_imu_to_camera_matrix(calib_in, matrix, args.second)
80-
81-
elif args.action == 'set_first_to_second':
86+
else:
87+
assert(args.action == 'set_first_to_second')
8288
calib_out = set_first_to_second_matrix(calib_in, matrix, args.second)
8389

8490
print(json.dumps(calib_out, indent=4))

0 commit comments

Comments
 (0)