Use env var to control max number of frames

blakeblackshear · Dec 1, 2024 · 36e5ab9 · 36e5ab9
1 parent 71e8f75
commit 36e5ab9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/frigate/app.py b/frigate/app.py
@@ -36,6 +36,7 @@
     EXPORT_DIR,
     MODEL_CACHE_DIR,
     RECORD_DIR,
+    SHM_FRAMES_VAR,
 )
 from frigate.db.sqlitevecq import SqliteVecQueueDatabase
 from frigate.embeddings import EmbeddingsContext, manage_embeddings
@@ -523,7 +524,10 @@ def shm_frame_count(self) -> int:
         if cam_total_frame_size == 0.0:
             return 0
 
-        shm_frame_count = min(200, int(available_shm / (cam_total_frame_size)))
+        shm_frame_count = min(
+            os.environ.get(SHM_FRAMES_VAR, 50),
+            int(available_shm / (cam_total_frame_size)),
+        )
 
         logger.debug(
             f"Calculated total camera size {available_shm} / {cam_total_frame_size} :: {shm_frame_count} frames for each camera in SHM"

diff --git a/frigate/const.py b/frigate/const.py
@@ -13,6 +13,8 @@
 PLUS_ENV_VAR = "PLUS_API_KEY"
 PLUS_API_HOST = "https://api.frigate.video"
 
+SHM_FRAMES_VAR = "SHM_MAX_FRAMES"
+
 # Attribute & Object constants
 
 DEFAULT_ATTRIBUTE_LABEL_MAP = {