fixed project name as argument

This commit is contained in:
2025-11-30 23:16:20 +01:00
parent 988b3e3807
commit 34df29bda5
+7 -2
View File
@@ -66,6 +66,7 @@ class Detector(abc.ABC):
filename = f"{RESULTS_FOLDER}/{det.get_board_name()}_cal.npz"
mtx, dist, r_vecs, t_vecs = det.calibration_get()
np.savez(f"{filename}", mtx=mtx, dist=dist, rvecs=r_vecs, tvecs=t_vecs)
print(f"Saved {filename} successfully!")
def calibrate_load(self):
for det in self.detector:
@@ -74,6 +75,7 @@ class Detector(abc.ABC):
with np.load(filename) as X:
mtx, dist, _, _ = [X[i] for i in ('mtx', 'dist', 'rvecs', 'tvecs')]
det.calibration_set(mtx, dist, None, None)
print(f"Loaded {filename} successfully!")
except FileNotFoundError:
pass
except KeyError:
@@ -183,6 +185,9 @@ def main():
ap.add_argument("-i", "--board_id", type=int,
default=0,
help="Boards id")
ap.add_argument("-p", "--name", type=str,
default='default',
help="Name of project")
args = vars(ap.parse_args())
msg_size = 3
@@ -191,12 +196,12 @@ def main():
if "Aruco" in args["type"]:
for n in range(args["num_boards"]):
board_id = f"{bid_start+n:0{msg_size}d}"
det = BoardDetectorAruco(f"Aruco_det-{n}", args["dict_type"], board_id, (args["board_width"], args["board_height"]))
det = BoardDetectorAruco(f"{args["name"]}-{n}", args["dict_type"], board_id, (args["board_width"], args["board_height"]))
board_detector.append(det)
elif "Charuco" in args["type"]:
for n in range(args["num_boards"]):
board_id = f"{bid_start+n:0{msg_size}d}"
det = BoardDetectorCharuco(f"Charuco_det-{n}", args["dict_type"], board_id, (args["board_width"], args["board_height"]))
det = BoardDetectorCharuco(f"{args["name"]}-{n}", args["dict_type"], board_id, (args["board_width"], args["board_height"]))
board_detector.append(det)
detector = Detector(args["source"], board_detector)