Fixed path formatting

This commit is contained in:
2025-12-27 11:30:29 +01:00
parent d3329e1e19
commit c7bb0c4812
2 changed files with 7 additions and 5 deletions

View File

@@ -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)
service.run_complete_detection(image, display=True)