fix(inference): resolve multiple YOLO inference and API issues

This commit is contained in:
2026-01-21 14:48:01 +08:00
parent 1b344aeb2e
commit 1c7190bbb0
5 changed files with 146 additions and 80 deletions

View File

@@ -23,16 +23,17 @@ class DatabaseConfig(BaseModel):
class ModelConfig(BaseModel):
engine_path: str = "models/yolo11s.engine"
onnx_path: str = "models/yolo11s.onnx"
pt_model_path: str = "models/yolo11s.pt"
engine_path: str = "models/yolo11n.engine"
onnx_path: str = "models/yolo11n.onnx"
pt_model_path: str = "models/yolo11n.pt"
imgsz: List[int] = [640, 640]
conf_threshold: float = 0.5
iou_threshold: float = 0.45
device: int = 0
batch_size: int = 8
half: bool = True
use_onnx: bool = True
half: bool = False
use_onnx: bool = False
use_trt: bool = False
class StreamConfig(BaseModel):