fixed project name as argument
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user