mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-04-05 22:57:40 +03:00
add audio filter config and load audio labels
This commit is contained in:
parent
46a6e34442
commit
262e0c4af9
@ -25,6 +25,7 @@ from frigate.plus import PlusApi
|
|||||||
from frigate.util.builtin import (
|
from frigate.util.builtin import (
|
||||||
deep_merge,
|
deep_merge,
|
||||||
get_ffmpeg_arg_list,
|
get_ffmpeg_arg_list,
|
||||||
|
load_labels,
|
||||||
)
|
)
|
||||||
from frigate.util.config import (
|
from frigate.util.config import (
|
||||||
CURRENT_CONFIG_VERSION,
|
CURRENT_CONFIG_VERSION,
|
||||||
@ -40,7 +41,7 @@ from frigate.util.services import auto_detect_hwaccel
|
|||||||
from .auth import AuthConfig
|
from .auth import AuthConfig
|
||||||
from .base import FrigateBaseModel
|
from .base import FrigateBaseModel
|
||||||
from .camera import CameraConfig, CameraLiveConfig
|
from .camera import CameraConfig, CameraLiveConfig
|
||||||
from .camera.audio import AudioConfig
|
from .camera.audio import AudioConfig, AudioFilterConfig
|
||||||
from .camera.birdseye import BirdseyeConfig
|
from .camera.birdseye import BirdseyeConfig
|
||||||
from .camera.detect import DetectConfig
|
from .camera.detect import DetectConfig
|
||||||
from .camera.ffmpeg import FfmpegConfig
|
from .camera.ffmpeg import FfmpegConfig
|
||||||
@ -671,6 +672,12 @@ class FrigateConfig(FrigateBaseModel):
|
|||||||
detector_config.model = model
|
detector_config.model = model
|
||||||
self.detectors[key] = detector_config
|
self.detectors[key] = detector_config
|
||||||
|
|
||||||
|
all_audio_labels = {
|
||||||
|
label
|
||||||
|
for label in load_labels("/audio-labelmap.txt", prefill=521).values()
|
||||||
|
if label
|
||||||
|
}
|
||||||
|
|
||||||
for name, camera in self.cameras.items():
|
for name, camera in self.cameras.items():
|
||||||
modified_global_config = global_config.copy()
|
modified_global_config = global_config.copy()
|
||||||
|
|
||||||
@ -791,6 +798,14 @@ class FrigateConfig(FrigateBaseModel):
|
|||||||
camera_config.review.genai.enabled
|
camera_config.review.genai.enabled
|
||||||
)
|
)
|
||||||
|
|
||||||
|
if camera_config.audio.filters is None:
|
||||||
|
camera_config.audio.filters = {}
|
||||||
|
|
||||||
|
audio_keys = all_audio_labels
|
||||||
|
audio_keys = audio_keys - camera_config.audio.filters.keys()
|
||||||
|
for key in audio_keys:
|
||||||
|
camera_config.audio.filters[key] = AudioFilterConfig()
|
||||||
|
|
||||||
# Add default filters
|
# Add default filters
|
||||||
object_keys = camera_config.objects.track
|
object_keys = camera_config.objects.track
|
||||||
if camera_config.objects.filters is None:
|
if camera_config.objects.filters is None:
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user