add FishEye calibration option

lambdaloop · Aug 3, 2019 · 7cf6fe9 · 7cf6fe9
1 parent 9175bba
commit 7cf6fe9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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)