ForestGuardAI/config.py at master · subeesesh/ForestGuardAI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
from dataclasses import dataclass, field
import os
from pathlib import Path
from typing import Dict, Tuple


@dataclass(frozen=True)
class Settings:
    project_name: str = "ForestGuardAI"
    root_dir: Path = field(default_factory=lambda: Path(__file__).resolve().parent)
    dataset_dir: Path = field(init=False)
    audio_dataset_dir: Path = field(init=False)
    models_dir: Path = field(init=False)
    audio_model_path: Path = field(init=False)
    yolo_model_path: Path = field(init=False)

    camera_index: int = 0
    camera_width: int = 640
    camera_height: int = 480
    inference_size: int = 416
    frame_skip: int = 5
    vision_confidence_threshold: float = 0.35

    audio_sample_rate: int = 16000
    audio_window_seconds: int = 2
    audio_channels: int = 1
    audio_n_mels: int = 64
    audio_confidence_threshold: float = 0.60
    audio_inference_interval: float = 0.50

    alert_cooldown_seconds: float = 3.0
    alert_history_size: int = 250
    pipeline_sleep_seconds: float = 0.05
    api_host: str = "127.0.0.1"
    api_port: int = 5000
    log_level: str = "INFO"

    audio_labels: Tuple[str, ...] = ("background", "chainsaw", "gunshot", "vehicle")
    vision_target_classes: Dict[int, str] = field(
        default_factory=lambda: {0: "person", 2: "car", 7: "truck"}
    )

    def __post_init__(self) -> None:
        object.__setattr__(self, "dataset_dir", self.root_dir / "dataset")
        dataset_override = os.getenv("FORESTGUARD_AUDIO_DATASET_DIR", "").strip()
        if dataset_override:
            object.__setattr__(self, "audio_dataset_dir", Path(dataset_override).expanduser().resolve())
        else:
            object.__setattr__(self, "audio_dataset_dir", self.dataset_dir / "audio")
        object.__setattr__(self, "models_dir", self.root_dir / "models")
        object.__setattr__(self, "audio_model_path", self.models_dir / "audio_model.pth")
        object.__setattr__(self, "yolo_model_path", self.models_dir / "yolov8n.pt")


settings = Settings()