diff --git a/ar_tag_pose/detector_board_main.py b/ar_tag_pose/detector_board_main.py index 1085488..6dd95b5 100644 --- a/ar_tag_pose/detector_board_main.py +++ b/ar_tag_pose/detector_board_main.py @@ -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)