fix(inference): resolve multiple YOLO inference and API issues
This commit is contained in:
11
config.py
11
config.py
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user