fix: изменение дирректории расположения весов модели

2 months ago · 01d8c8f8de
parent 86dad9b05b
commit 01d8c8f8de
2 changed files with 1 additions and 1 deletions
--- a/agentic/real-sense-object-detection-and-track-service/Interfaces/DataProcessor.py
+++ b/agentic/real-sense-object-detection-and-track-service/Interfaces/DataProcessor.py
@ -3,7 +3,7 @@ from ultralytics import YOLO
 import numpy as np

 class DataProcessor:
-    def __init__(self, model_path='./models/yolov9e_object_classification.pt'):
+    def __init__(self, model_path='./../models/yolov9e_object_classification.pt'):
        self.model = YOLO(model_path)

    def __decode_image_from_base64(self, data, color=True):
--- a/agentic/real-sense-object-detection-and-track-service/models/yolov9e_object_classification.pt
+++ b/agentic/real-sense-object-detection-and-track-service/models/yolov9e_object_classification.pt