diff options
Diffstat (limited to 'src/argaze/ArUcoMarkers/ArUcoCamera.py')
-rw-r--r-- | src/argaze/ArUcoMarkers/ArUcoCamera.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/argaze/ArUcoMarkers/ArUcoCamera.py b/src/argaze/ArUcoMarkers/ArUcoCamera.py index 5efc4f8..43a7cf5 100644 --- a/src/argaze/ArUcoMarkers/ArUcoCamera.py +++ b/src/argaze/ArUcoMarkers/ArUcoCamera.py @@ -210,11 +210,12 @@ class ArUcoCamera(ArFeatures.ArCamera): # Return dection time and exceptions return detection_time, exceptions - def __image(self, draw_detected_markers: dict = None, **kwargs) -> numpy.array: + def __image(self, draw_detected_markers: dict = None, draw_optic_parameters_grid: dict = None, **kwargs) -> numpy.array: """Get frame image with ArUco detection visualisation. Parameters: draw_detected_markers: ArucoMarker.draw parameters (if None, no marker drawn) + draw_optic_parameters_grid: OpticParameter.draw parameters (if None, no grid drawn) kwargs: ArCamera.image parameters """ @@ -233,6 +234,11 @@ class ArUcoCamera(ArFeatures.ArCamera): self.aruco_detector.draw_detected_markers(image, draw_detected_markers) + # Draw optic parameters grid if required + if draw_optic_parameters_grid is not None: + + self.aruco_detector.optic_parameters.draw(image, **draw_optic_parameters_grid) + # Unlock camera frame exploitation self._frame_lock.release() |