Defing PipelineStepObject and PipelineStepMethod to assess time execution. Removing exception managment to let the user catch them into the script.

1 files changed, 3 insertions, 1 deletions

diff --git a/src/argaze/ArUcoMarkers/ArUcoDetector.py b/src/argaze/ArUcoMarkers/ArUcoDetector.py
index e6a305f..6d12f3d 100644
--- a/src/argaze/ArUcoMarkers/ArUcoDetector.py
+++ b/src/argaze/ArUcoMarkers/ArUcoDetector.py
@@ -14,6 +14,7 @@ import os
 from collections import Counter
 import time
 
+from argaze import DataFeatures
 from argaze.ArUcoMarkers import ArUcoMarkersDictionary, ArUcoMarker, ArUcoOpticCalibrator
 
 import numpy
@@ -131,7 +132,7 @@ class DetectorParameters():
         return self.__parameters
 
 @dataclass
-class ArUcoDetector():
+class ArUcoDetector(DataFeatures.PipelineStepObject):
     """ArUco markers detector.
 
     Parameters:
@@ -255,6 +256,7 @@ class ArUcoDetector():
 
         return output
 
+    @DataFeatures.PipelineStepMethod
     def detect_markers(self, image: numpy.array) -> float:
         """Detect all ArUco markers into an image.