From ca5ea81397c14d7e0d5b2c378768e67a87e517b2 Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Fri, 28 Nov 2025 19:04:30 +0100 Subject: [PATCH] create results folder if not exists --- ar_tag_pose/ar_tag_pose.py | 6 ++++++ ar_tag_pose/ar_tag_pose_aruco_board.py | 1 + ar_tag_pose/ar_tag_pose_charuco_board.py | 1 + 3 files changed, 8 insertions(+) diff --git a/ar_tag_pose/ar_tag_pose.py b/ar_tag_pose/ar_tag_pose.py index 31505c5..54b126b 100644 --- a/ar_tag_pose/ar_tag_pose.py +++ b/ar_tag_pose/ar_tag_pose.py @@ -2,8 +2,11 @@ import abc import math import cv2 as cv import numpy as np +from pathlib import Path from utils import to_board_pose_euler +RESULTS_FOLDER = "results" + class ArTagPose(abc.ABC): detector: cv.aruco.ArucoDetector | cv.aruco.CharucoDetector = None board: cv.aruco.GridBoard | cv.aruco.CharucoBoard = None @@ -15,6 +18,9 @@ class ArTagPose(abc.ABC): obj_points_list = [] img_points_list = [] + def __init__(self): + Path.mkdir(Path(RESULTS_FOLDER), exist_ok=True) + @staticmethod def cap_init(cap: cv.VideoCapture): cap.set(cv.CAP_PROP_SETTINGS, 1) diff --git a/ar_tag_pose/ar_tag_pose_aruco_board.py b/ar_tag_pose/ar_tag_pose_aruco_board.py index d148efb..43f303f 100644 --- a/ar_tag_pose/ar_tag_pose_aruco_board.py +++ b/ar_tag_pose/ar_tag_pose_aruco_board.py @@ -15,6 +15,7 @@ def factory(dict_type: int, board_size: cv.typing.Size, marker_length: float=100 class ArTagPoseArucoBoard(ArTagPose): def __init__(self, dict_type: int, board_size: tuple[int, int]): + ArTagPose.__init__(self) self.board_size = board_size self.detector, self.board, self.name = factory(dict_type, board_size) diff --git a/ar_tag_pose/ar_tag_pose_charuco_board.py b/ar_tag_pose/ar_tag_pose_charuco_board.py index 305ae6c..87ac8ad 100644 --- a/ar_tag_pose/ar_tag_pose_charuco_board.py +++ b/ar_tag_pose/ar_tag_pose_charuco_board.py @@ -15,6 +15,7 @@ def factory(dict_type: int, board_size: cv.typing.Size, square_length: float = 5 class ArTagPoseCharucoBoard(ArTagPose): def __init__(self, dict_type: int, board_size: tuple[int, int]): + ArTagPose.__init__(self) self.board_size = board_size self.detector, self.board, self.name = factory(dict_type, board_size)