配置修改

This commit is contained in:
2026-01-21 13:29:00 +08:00
parent 1e562798eb
commit e965b10603
4 changed files with 58 additions and 24 deletions

View File

@@ -23,14 +23,16 @@ class DatabaseConfig(BaseModel):
class ModelConfig(BaseModel):
engine_path: str = "models/yolo11n_fp16_480.engine"
pt_model_path: str = "models/yolo11n.pt"
imgsz: List[int] = [480, 480]
engine_path: str = "models/yolo11s.engine"
onnx_path: str = "models/yolo11s.onnx"
pt_model_path: str = "models/yolo11s.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
class StreamConfig(BaseModel):
@@ -78,6 +80,17 @@ class LoggingConfig(BaseModel):
backup_count: int = 5
class CloudConfig(BaseModel):
enabled: bool = False
api_url: str = "https://api.example.com"
api_key: str = ""
device_id: str = "EDGE-001"
sync_interval: int = 60
alarm_retry_interval: int = 60
status_report_interval: int = 60
max_retries: int = 3
class MonitoringConfig(BaseModel):
enabled: bool = True
port: int = 9090
@@ -93,6 +106,7 @@ class LLMConfig(BaseModel):
class Config(BaseModel):
cloud: CloudConfig = Field(default_factory=CloudConfig)
database: DatabaseConfig = Field(default_factory=DatabaseConfig)
model: ModelConfig = Field(default_factory=ModelConfig)
stream: StreamConfig = Field(default_factory=StreamConfig)