diff --git a/ar_tag_pose/detector_board_main.py b/ar_tag_pose/detector_board_main.py index 6b4da8e..38b5bfc 100644 --- a/ar_tag_pose/detector_board_main.py +++ b/ar_tag_pose/detector_board_main.py @@ -70,13 +70,14 @@ class Detector(abc.ABC): return f"{RESULTS_FOLDER}/{self.camera_name}_cal.npz" def _get_params_filename(self): - return f"{RESULTS_FOLDER}/{self.params["name"]}.json" + return f"{RESULTS_FOLDER}/{self.name}.json" def _get_camera_name(self): - camera_name = "default" - if self.cap is not None: - backends = {'V4L2': cv.CAP_V4L} - cameras = enumerate_cameras(backends[self.cap.getBackendName()]) + camera_name = self.name + backend_name = self.cap.getBackendName() + if 'V4L2' in backend_name: + backends = {backend_name: cv.CAP_V4L} + cameras = enumerate_cameras(backends[backend_name]) for cam in cameras: if cam.index == self.device_id: