fulyvv/pose2sim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Check for consistent camera number

Browse Source
This commit is contained in:
David PAGNON 2024-02-19 13:13:33 +01:00 committed by GitHub
parent a8bb21b068
commit 128730a334
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
Pose2Sim/calibration.py
View File

@ -504,11 +504,26 @@ def calib_calc_fun(calib_dir, intrinsics_config_dict, extrinsics_config_dict):
D += [calib_data[cam]['distortions']] D += [calib_data[cam]['distortions']]
R += [[0.0, 0.0, 0.0]] R += [[0.0, 0.0, 0.0]]
T += [[0.0, 0.0, 0.0]] T += [[0.0, 0.0, 0.0]]
nb_cams_intrinsics = len(C)
# calculate intrinsics otherwise # calculate intrinsics otherwise
else: else:
logging.info(f'\nCalculating intrinsic parameters...') logging.info(f'\nCalculating intrinsic parameters...')
ret, C, S, D, K, R, T = calibrate_intrinsics(calib_dir, intrinsics_config_dict) ret, C, S, D, K, R, T = calibrate_intrinsics(calib_dir, intrinsics_config_dict)
nb_cams_intrinsics = len(C)
# calculate extrinsics
if calculate_extrinsics:
logging.info(f'\nCalculating extrinsic parameters...')
# check that the number of cameras is consistent
nb_cams_extrinsics = len(next(os.walk(os.path.join(calib_dir, 'extrinsics')))[1])
if nb_cams_intrinsics != nb_cams_extrinsics:
raise Exception(f'Error: The number of cameras is not consistent:\
Found {nb_cams_intrinsics} cameras based on the number of intrinsic folders or on calibration file data,\
and {nb_cams_extrinsics} cameras from based on the number of extrinsic folders.')
ret, C, S, D, K, R, T = calibrate_intrinsics(calib_dir, intrinsics_config_dict)
# calculate extrinsics # calculate extrinsics
if calculate_extrinsics: if calculate_extrinsics: