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

Refactor audio capture and processing threads to use more descriptive thread names

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

13
frigate/audio.py
View File

@ -10,10 +10,9 @@ import threading
import numpy as np import numpy as np
from setproctitle import setproctitle from setproctitle import setproctitle
from frigate.config import CameraConfig, AudioModelConfig from frigate.config import AudioModelConfig, CameraConfig
from frigate.object_detection import RemoteObjectDetector from frigate.object_detection import RemoteObjectDetector
from frigate.util import listen, SharedMemoryFrameManager from frigate.util import SharedMemoryFrameManager, listen
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -31,8 +30,8 @@ def capture_audio(
signal.signal(signal.SIGTERM, receiveSignal) signal.signal(signal.SIGTERM, receiveSignal)
signal.signal(signal.SIGINT, receiveSignal) signal.signal(signal.SIGINT, receiveSignal)
threading.current_thread().name = f"capture:{name}" threading.current_thread().name = f"capture_audio:{name}"
setproctitle(f"frigate.capture:{name}") setproctitle(f"frigate.capture_audio:{name}")
listen() listen()
chunk_size = int(round(model_config.duration * model_config.sample_rate * 2)) chunk_size = int(round(model_config.duration * model_config.sample_rate * 2))
@ -84,8 +83,8 @@ def process_audio(
signal.signal(signal.SIGTERM, receiveSignal) signal.signal(signal.SIGTERM, receiveSignal)
signal.signal(signal.SIGINT, receiveSignal) signal.signal(signal.SIGINT, receiveSignal)
threading.current_thread().name = f"process:{name}" threading.current_thread().name = f"process_audio:{name}"
setproctitle(f"frigate.process:{name}") setproctitle(f"frigate.process_audio:{name}")
listen() listen()
shape = (int(round(model_config.duration * model_config.sample_rate)),) shape = (int(round(model_config.duration * model_config.sample_rate)),)

2
frigate/events/audio.py
View File

@ -2,13 +2,13 @@
import logging import logging
import multiprocessing as mp import multiprocessing as mp
import numpy as np
import signal import signal
import subprocess as sp import subprocess as sp
import threading import threading
from types import FrameType from types import FrameType
from typing import Optional from typing import Optional
import numpy as np
from setproctitle import setproctitle from setproctitle import setproctitle
from frigate.config import CameraConfig, FrigateConfig from frigate.config import CameraConfig, FrigateConfig