aboutsummaryrefslogtreecommitdiff
path: root/src/argaze/ArUcoMarkers/ArUcoDetector.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/argaze/ArUcoMarkers/ArUcoDetector.py')
-rw-r--r--src/argaze/ArUcoMarkers/ArUcoDetector.py4
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.