diff options
| -rw-r--r-- | src/argaze/utils/tobii_camera_calibrate.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/src/argaze/utils/tobii_camera_calibrate.py b/src/argaze/utils/tobii_camera_calibrate.py index 65b177e..f19cb8b 100644 --- a/src/argaze/utils/tobii_camera_calibrate.py +++ b/src/argaze/utils/tobii_camera_calibrate.py @@ -49,6 +49,10 @@ def main(): # Setup camera at 25 fps to work on Full HD video stream tobii_controller.set_scene_camera_freq_25() + # Get video stream dimension + video_width = tobii_controller.get_configuration()['sys_sc_width'] + video_height = tobii_controller.get_configuration()['sys_sc_height'] + # Print current confirugration print(f'Tobii Glasses Pro 2 configuration:') for key, value in tobii_controller.get_configuration().items(): @@ -64,7 +68,8 @@ def main(): aruco_board = ArUcoBoard.ArUcoBoard(args.dictionary, args.columns, args.rows, args.square_size, args.marker_size) # Create aruco detecter - aruco_detector = ArUcoDetector.ArUcoDetector(args.dictionary, args.marker_size, aruco_camera) + config = {'dictionary':'DICT_APRILTAG_16h5', 'marker_size': 5, 'camera':{}, 'parameters':{}} + aruco_detector = ArUcoDetector.ArUcoDetector(**config) # Start tobii glasses streaming tobii_controller.start_streaming() @@ -119,15 +124,15 @@ def main(): tobii_controller.stop_streaming() print('\nCalibrating camera...') - aruco_camera.calibrate(aruco_board, video_frame.width, video_frame.height) + aruco_camera.calibrate(aruco_board, video_width, video_height) print('\nCalibration succeeded!') print(f'\nRMS:\n{aruco_camera.rms}') - print(f'\nDimensions:\n{video_frame.width}x{video_frame.height}') + print(f'\nDimensions:\n{video_width}x{video_height}') print(f'\nCamera matrix:\n{aruco_camera.K}') print(f'\nDistortion coefficients:\n{aruco_camera.D}') - aruco_camera.save_calibration_file(args.output) + aruco_camera.to_json(args.output) print(f'\nCalibration data exported into {args.output} file') |