aboutsummaryrefslogtreecommitdiff
path: root/src/argaze/ArUcoMarkers/ArUcoCamera.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/argaze/ArUcoMarkers/ArUcoCamera.py')
-rw-r--r--src/argaze/ArUcoMarkers/ArUcoCamera.py29
1 files changed, 16 insertions, 13 deletions
diff --git a/src/argaze/ArUcoMarkers/ArUcoCamera.py b/src/argaze/ArUcoMarkers/ArUcoCamera.py
index aa7426c..b63743d 100644
--- a/src/argaze/ArUcoMarkers/ArUcoCamera.py
+++ b/src/argaze/ArUcoMarkers/ArUcoCamera.py
@@ -48,19 +48,22 @@ class ArUcoCamera():
json.dump(calibration_data, calibration_file, ensure_ascii=False, indent=4)
- def get_rms(self):
- """Get Root Mean Square (rms) error.
- **Returns:** float"""
+ @property
+ def rms(self) -> float:
+ """Get Root Mean Square (rms) error."""
+
return self.__rms
- def get_K(self):
- """Get camera matrix.
- **Returns:** numpy.array"""
+ @property
+ def K(self) -> numpy.array:
+ """Get camera matrix."""
+
return self.__K
- def get_D(self):
- """Get camera distorsion coefficients.
- **Returns:** numpy.array"""
+ @property
+ def D(self) -> numpy.array:
+ """Get camera distorsion coefficients."""
+
return self.__D
def calibrate(self, board, frame_width, frame_height):
@@ -69,7 +72,7 @@ class ArUcoCamera():
if self.__corners_set_number > 0:
self.__dimensions = [frame_width, frame_height]
- self.__rms, self.__K, self.__D, r, t = aruco.calibrateCameraCharuco(self.__corners_set, self.__corners_set_ids, board.get_model(), self.__dimensions, None, None)
+ self.__rms, self.__K, self.__D, r, t = aruco.calibrateCameraCharuco(self.__corners_set, self.__corners_set_ids, board.model, self.__dimensions, None, None)
def reset_calibration_data(self):
"""Clear all calibration data."""
@@ -85,9 +88,9 @@ class ArUcoCamera():
self.__corners_set.append(corners)
self.__corners_set_ids.append(corners_ids)
- def get_calibration_data_count(self):
- """Get how much calibration data are stored.
- **Returns:** int"""
+ @property
+ def calibration_data_count(self) -> int:
+ """Get how much calibration data are stored."""
return self.__corners_set_number