Fixed path formatting
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user