administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-05 10:45:21 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add AudioModelConfig to detector imports and audio configuration options

Browse Source
This commit is contained in:
Sergey Krashevich 2023-06-19 11:10:14 +03:00
parent 3bd46db1ae
commit da19af3c35
No known key found for this signature in database
GPG Key ID: 625171324E7D3856
3 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/config.py
View File

@ -12,7 +12,7 @@ from pydantic import BaseModel, Extra, Field, parse_obj_as, validator
from pydantic.fields import PrivateAttr
from frigate.const import CACHE_DIR, DEFAULT_DB_PATH, REGEX_CAMERA_NAME, YAML_EXT
from frigate.detectors import DetectorConfig, ModelConfig
from frigate.detectors import AudioModelConfig, DetectorConfig, ModelConfig
from frigate.detectors.detector_config import BaseDetectorConfig
from frigate.ffmpeg_presets import (
parse_preset_hardware_acceleration_decode,

7
frigate/detectors/__init__.py
View File

@ -1,6 +1,11 @@
import logging
from .detector_config import InputTensorEnum, ModelConfig, PixelFormatEnum # noqa: F401
from .detector_config import ( # noqa: F401
AudioModelConfig,
InputTensorEnum,
ModelConfig,
PixelFormatEnum,
)
from .detector_types import DetectorConfig, DetectorTypeEnum, api_types # noqa: F401
logger = logging.getLogger(__name__)

13
frigate/detectors/detector_config.py
View File

@ -10,6 +10,7 @@ import requests
from pydantic import BaseModel, Extra, Field
from pydantic.fields import PrivateAttr
from frigate.const import AUDIO_DURATION, AUDIO_FORMAT, AUDIO_SAMPLE_RATE
from frigate.plus import PlusApi
from frigate.util import load_labels
@ -39,6 +40,18 @@ class ModelTypeEnum(str, Enum):
yolov8 = "yolov8"
class AudioModelConfig(BaseModel):
path: str = Field(
default="/cpu_audio_model.tflite", title="Custom Object detection model path."
)
labelmap_path: str = Field(
default="/audio-labelmap.txt", title="Label map for custom object detector."
)
duration: float = Field(default=AUDIO_DURATION, title="Audio duration.")
format: str = Field(default=AUDIO_FORMAT, title="Audio format.")
sample_rate: int = Field(default=AUDIO_SAMPLE_RATE, title="Audio sample rate.")
class ModelConfig(BaseModel):
path: Optional[str] = Field(title="Custom Object detection model path.")
labelmap_path: Optional[str] = Field(title="Label map for custom object detector.")