From e072788f3586437d91258e142fab0a5ec7ca9825 Mon Sep 17 00:00:00 2001
From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com>
Date: Fri, 15 Apr 2022 09:00:11 -0500
Subject: [PATCH] Add class variable and update in process_frames

---
 frigate/motion.py | 5 +++--
 frigate/video.py  | 4 +++-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/frigate/motion.py b/frigate/motion.py
index de8440668..d426d681d 100644
--- a/frigate/motion.py
+++ b/frigate/motion.py
@@ -24,8 +24,9 @@ class MotionDetector:
         )
         self.mask = np.where(resized_mask == [0])
         self.save_images = False
+        self.improve_contrast = self.config.improve_contrast
 
-    def detect(self, frame, improve_contrast):
+    def detect(self, frame):
         motion_boxes = []
 
         gray = frame[0 : self.frame_shape[0], 0 : self.frame_shape[1]]
@@ -38,7 +39,7 @@ class MotionDetector:
         )
 
         # Improve contrast
-        if improve_contrast:
+        if self.improve_contrast:
             minval = np.percentile(resized_frame, 4)
             maxval = np.percentile(resized_frame, 96)
             # don't adjust if the image is a single color
diff --git a/frigate/video.py b/frigate/video.py
index 5f46dba8c..b3e40fbdb 100755
--- a/frigate/video.py
+++ b/frigate/video.py
@@ -494,8 +494,10 @@ def process_frames(
             logger.info(f"{camera_name}: frame {frame_time} is not in memory store.")
             continue
 
+        motion_detector.improve_contrast = improve_contrast_enabled.value
+
         # look for motion
-        motion_boxes = motion_detector.detect(frame, improve_contrast_enabled.value)
+        motion_boxes = motion_detector.detect(frame)
 
         regions = []