improved naming of files

This commit is contained in:
2025-11-28 14:49:15 +01:00
parent 14cb97a8f3
commit 1f0af82a6e
+3 -1
View File
@@ -1,6 +1,7 @@
from PIL.ImageOps import posterize
from ar_tag_pose import ArTagPose
from aruco_types import ARUCO_DICT
import numpy as np
import cv2 as cv
@@ -12,7 +13,8 @@ def factory(dict_type: int, board_size: cv.typing.Size, marker_length: float=100
im_w = int(board_size[0] * (marker_length + marker_sep) - marker_sep + 2 * margins)
im_h = int(board_size[1] * (marker_length + marker_sep) - marker_sep + 2 * margins)
image = board.generateImage((im_w, im_h), None, margins, 1)
cv.imwrite("results/aruco_board.png", image)
keyname = [key for key, val in ARUCO_DICT.items() if val == dict_type][0]
cv.imwrite(f"results/aruco_board_w{board_size[0]}_h{board_size[1]}_{keyname}.png", image)
return detector, board