Improved the chessboard detection

commit-id:6662bc18
child-lab-uj · Jan 19, 2025 · 3412084 · 3412084
1 parent 1e9c2f1
commit 3412084
Show file tree

Hide file tree

Showing 3 changed files with 558 additions and 55 deletions.
diff --git a/child_lab_framework/_cli/cli.py b/child_lab_framework/_cli/cli.py
@@ -89,7 +89,7 @@ def calibrate(
         calibration = calibration_procedure.run(
             video_input,
             video_output,
-            chessboard.Properties(square_size, *inner_board_corners),
+            chessboard.BoardProperties(square_size, *inner_board_corners),
             skip,
         )
 

diff --git a/child_lab_framework/_procedure/calibrate.py b/child_lab_framework/_procedure/calibrate.py
@@ -16,7 +16,7 @@
 def run(
     video_source: Path,
     annotated_video_destination: Path,
-    board_properties: chessboard.Properties,
+    board_properties: chessboard.BoardProperties,
     skip: int,
 ) -> Calibration:
     reader = Reader(
@@ -39,8 +39,8 @@ def run(
 
     detector = chessboard.Detector(board_properties)
 
-    inner_corners_per_row = board_properties.inner_corners_per_row
-    inner_corners_per_column = board_properties.inner_corners_per_column
+    inner_corners_per_row = board_properties.inner_columns
+    inner_corners_per_column = board_properties.inner_rows
     square_size = board_properties.square_size
 
     # Prepare object points, like (0,0,0), (1,0,0), (2,0,0). ...,(6,5,0)