Naaturel/board-mate
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed path formatting

Browse Source
This commit is contained in:
Laurent
2025-12-27 11:30:29 +01:00
parent d3329e1e19
commit c7bb0c4812
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rpi/main.py
View File

@@ -25,8 +25,6 @@ def detect() :
print("Error: Could not open camera") print("Error: Could not open camera")
exit() exit()
cv2.namedWindow("Predictions", cv2.WINDOW_NORMAL)
while True: while True:
ret, frame = cap.read() ret, frame = cap.read()
fen = detection_service.get_fen(frame) fen = detection_service.get_fen(frame)

8
rpi/services/detection_service.py
View File

@@ -1,5 +1,6 @@
import cv2 import cv2
import numpy as np import numpy as np
from pathlib import Path
from ultralytics.engine.results import Results from ultralytics.engine.results import Results
from models.detection.detector import Detector from models.detection.detector import Detector
@@ -18,8 +19,11 @@ class DetectionService:
scale_size : tuple[int, int] scale_size : tuple[int, int]
def __init__(self): def __init__(self):
self.edges_detector = Detector("../assets/models/edges.pt") current_file = Path(__file__).resolve()
self.pieces_detector = Detector("../assets/models/unified-nano-refined.pt") 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.pieces_manager = PiecesManager()
self.board_manager = BoardManager() self.board_manager = BoardManager()