diff options
Diffstat (limited to 'src/argaze/ArUcoMarkers/ArUcoCamera.py')
-rw-r--r-- | src/argaze/ArUcoMarkers/ArUcoCamera.py | 29 |
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 |