upload model to cdn and api

switch to yolov11
This commit is contained in:
Alex Bezdieniezhnykh
2025-03-03 23:36:10 +02:00
parent ceb50bf48a
commit 2fa864018f
14 changed files with 258 additions and 86 deletions
+27
View File
@@ -1,6 +1,33 @@
from abc import ABC, abstractmethod
from ultralytics import YOLO
import yaml
class Predictor(ABC):
@abstractmethod
def predict(self, frame):
pass
class OnnxPredictor(Predictor):
def __init__(self):
self.model = YOLO('azaion.onnx')
self.model.task = 'detect'
with open('data.yaml', 'r') as f:
data_yaml = yaml.safe_load(f)
class_names = data_yaml['names']
names = self.model.names
def predict(self, frame):
results = self.model.track(frame, persist=True, tracker='bytetrack.yaml')
return results[0].plot()
class YoloPredictor(Predictor):
def __init__(self):
self.model = YOLO('azaion.pt')
def predict(self, frame):
results = self.model.track(frame, persist=True, tracker='bytetrack.yaml')
return results[0].plot()