diff --git a/anipose/anipose.py b/anipose/anipose.py index c5de8eb..bd057be 100644 --- a/anipose/anipose.py +++ b/anipose/anipose.py @@ -11,7 +11,8 @@ 'video_extension': 'avi', 'calibration': { 'animal_calibration': False, - 'calibration_init': None + 'calibration_init': None, + 'fisheye': False }, 'triangulation': { 'ransac': False, diff --git a/anipose/calibrate.py b/anipose/calibrate.py index 1d7e247..ba40461 100644 --- a/anipose/calibrate.py +++ b/anipose/calibrate.py @@ -145,10 +145,11 @@ def process_session(config, session_path): error = None elif config['calibration']['calibration_init'] is not None: calib_path = os.path.join(config['path'], config['calibration']['calibration_init']) + print('loading calibration from: {}'.format(calib_path)) cgroup = CameraGroup.load(calib_path) init_extrinsics = False else: - cgroup = CameraGroup.from_names(cam_names) + cgroup = CameraGroup.from_names(cam_names, config['calibration']['fisheye']) board = get_calibration_board(config)