create results folder if not exists

This commit is contained in:
2025-11-28 19:04:30 +01:00
parent 6e30df20b6
commit ca5ea81397
3 changed files with 8 additions and 0 deletions
+6
View File
@@ -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)
+1
View File
@@ -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)
+1
View File
@@ -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)