|
|
@ -3,7 +3,7 @@ from ultralytics import YOLO
|
|
|
|
import numpy as np
|
|
|
|
import numpy as np
|
|
|
|
|
|
|
|
|
|
|
|
class DataProcessor:
|
|
|
|
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)
|
|
|
|
self.model = YOLO(model_path)
|
|
|
|
|
|
|
|
|
|
|
|
def __decode_image_from_base64(self, data, color=True):
|
|
|
|
def __decode_image_from_base64(self, data, color=True):
|
|
|
|