Merge branch '23-return-params'

SciKit-Surgery · Jul 28, 2020 · 25d13ad · 25d13ad
2 parents 63405ae + 56b0651
commit 25d13ad
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/sksurgerycalibration/video/video_calibration_driver_mono.py b/sksurgerycalibration/video/video_calibration_driver_mono.py
@@ -183,4 +183,4 @@ def handeye_calibration(self):
 
         self.calibration_params.set_handeye(handeye, pattern2marker)
 
-        return proj_err, recon_err
+        return proj_err, recon_err, copy.deepcopy(self.calibration_params)
diff --git a/sksurgerycalibration/video/video_calibration_driver_stereo.py b/sksurgerycalibration/video/video_calibration_driver_stereo.py
@@ -258,4 +258,4 @@ def handeye_calibration(self):
         self.calibration_params.right_params.set_handeye(
             r_handeye, r_pattern2marker)
 
-        return proj_err, recon_err
+        return proj_err, recon_err, copy.deepcopy(self.calibration_params)
diff --git a/tests/video/test_hand_eye.py b/tests/video/test_hand_eye.py
@@ -91,7 +91,7 @@ def test_handeye_calibration_mono():
     assert reproj_err_1 < one_pixel
     assert recon_err_1 < one_pixel
 
-    proj_err, recon_err = calibrator.handeye_calibration()
+    proj_err, recon_err, _ = calibrator.handeye_calibration()
 
     print(f'Reproj err {proj_err}')
     print(f'Recon err {recon_err}')
@@ -152,7 +152,7 @@ def test_handeye_calibration_stereo():
     assert reproj_err_1 < one_pixel
     assert recon_err_1 < one_pixel
 
-    proj_err, recon_err = calibrator.handeye_calibration()
+    proj_err, recon_err, _ = calibrator.handeye_calibration()
 
     print(f'Reproj err {proj_err}')
     print(f'Recon err {recon_err}')