Merge pull request #198 from zivid/2025-03-17-update-python-samples

Samples: Update capture settings and calibration board detection

Author: Jorgen-ha

modules/zividsamples/white_balance_calibration.py

@@ -103,7 +103,7 @@ def white_balance_calibration(
         settings_2d.processing.color.balance.green = corrected_green_balance
         settings_2d.processing.color.balance.blue = corrected_blue_balance
 
-        rgba = camera.capture(settings_2d).image_rgba().copy_data()
+        rgba = camera.capture_2d(settings_2d).image_rgba().copy_data()
         mean_color = compute_mean_rgb_from_mask(rgba[:, :, 0:3], mask)
 
         mean_red, mean_green, mean_blue = mean_color[0], mean_color[1], mean_color[2]

source/applications/advanced/auto_2d_settings.py

@@ -27,6 +27,7 @@
 import matplotlib.pyplot as plt
 import numpy as np
 import zivid
+import zivid.calibration
 from zividsamples.calibration_board_utils import find_white_mask_from_checkerboard
 from zividsamples.white_balance_calibration import (
     camera_may_need_color_balancing,
@@ -111,30 +112,10 @@ def _capture_rgb(camera: zivid.Camera, settings_2d: zivid.Settings2D) -> np.ndar
         rgb: RGB image (H, W, 3)
 
     """
-    rgb = camera.capture(settings_2d).image_rgba().copy_data()[:, :, :3]
+    rgb = camera.capture_2d(settings_2d).image_rgba().copy_data()[:, :, :3]
     return rgb
 
 
-def _capture_assistant_settings(camera: zivid.Camera) -> zivid.Settings:
-    """Get settings from capture assistant.
-
-    Args:
-        camera: Zivid camera
-
-    Returns:
-        Zivid 3D capture settings from capture assistant
-
-    """
-    suggest_settings_parameters = zivid.capture_assistant.SuggestSettingsParameters(
-        max_capture_time=timedelta(milliseconds=1200),
-        ambient_light_frequency=zivid.capture_assistant.SuggestSettingsParameters.AmbientLightFrequency.none,
-    )
-    suggested_settings = zivid.capture_assistant.suggest_settings(camera, suggest_settings_parameters)
-    suggested_settings.sampling.pixel = "all"
-
-    return suggested_settings
-
-
 def _find_white_mask_and_distance_to_checkerboard(camera: zivid.Camera) -> Tuple[np.ndarray, float]:
     """Generate a 2D mask of the white checkers on a checkerboard and calculate the distance to it.
 
@@ -150,8 +131,7 @@ def _find_white_mask_and_distance_to_checkerboard(camera: zivid.Camera) -> Tuple
 
     """
     try:
-        settings = _capture_assistant_settings(camera)
-        frame = camera.capture(settings)
+        frame = zivid.calibration.capture_calibration_board(camera)
 
         checkerboard_pose = zivid.calibration.detect_calibration_board(frame).pose().to_matrix()
         distance_to_checkerboard = checkerboard_pose[2, 3]