diff --git a/rpi/main.py b/rpi/main.py index 56cdb3ca..a474514a 100644 --- a/rpi/main.py +++ b/rpi/main.py @@ -25,8 +25,6 @@ def detect() : print("Error: Could not open camera") exit() - cv2.namedWindow("Predictions", cv2.WINDOW_NORMAL) - while True: ret, frame = cap.read() fen = detection_service.get_fen(frame) diff --git a/rpi/services/detection_service.py b/rpi/services/detection_service.py index e6071dbe..efd27e94 100644 --- a/rpi/services/detection_service.py +++ b/rpi/services/detection_service.py @@ -1,5 +1,6 @@ import cv2 import numpy as np +from pathlib import Path from ultralytics.engine.results import Results from models.detection.detector import Detector @@ -18,8 +19,11 @@ class DetectionService: scale_size : tuple[int, int] def __init__(self): - self.edges_detector = Detector("../assets/models/edges.pt") - self.pieces_detector = Detector("../assets/models/unified-nano-refined.pt") + current_file = Path(__file__).resolve() + project_root = current_file.parent.parent + + self.edges_detector = Detector(project_root / "assets" / "models" / "edges.pt") + self.pieces_detector = Detector(project_root / "assets" / "models" / "unified-nano-refined.pt") self.pieces_manager = PiecesManager() self.board_manager = BoardManager() @@ -96,4 +100,4 @@ if __name__ == "__main__" : fen = service.get_fen(image) print(fen) - service.run_complete_detection(image, display=True) \ No newline at end of file + service.run_complete_detection(image, display=True)