配置修改
This commit is contained in:
20
config.py
20
config.py
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user