get charuco chessboard pts if using charuco

ros-perception · Apr 28, 2024 · 6834a56 · 6834a56
1 parent d3c7fd0
commit 6834a56
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/camera_calibration/src/camera_calibration/calibrator.py b/camera_calibration/src/camera_calibration/calibrator.py
@@ -476,8 +476,11 @@ def compute_goodenough(self):
         return list(zip(self._param_names, min_params, max_params, progress))
 
     def mk_object_points(self, boards, use_board_size = False):
+        if self.pattern == Patterns.ChArUco:
+            opts = [board.charuco_board.chessboardCorners for board in boards]
+            return opts
         opts = []
-        for i, b in enumerate(boards):
+        for b in boards:
             num_pts = b.n_cols * b.n_rows
             opts_loc = numpy.zeros((num_pts, 1, 3), numpy.float32)
             for j in range(num_pts):