chris/frigate
1
0
Fork 0
forked from Github/frigate
Code Pull Requests Packages Activity

Allow improve_contrast to be toggled via mqtt (#3011)

Browse Source 
* Toggle improve_contrast for cameras via MQTT

* Process parameter to mqtt toggle improve_contrast

* Update mqtt docs for improve_contrast topic

* Spacing

* Add class variable and update in process_frames

* Pass to constructor

* pass by reference mistake

* remove parameter

* remove parameter
This commit is contained in:
Josh Hawkins
2022-04-16 08:52:02 -05:00
committed by GitHub
parent a5016afdd4
commit 65e0ec7826
5 changed files with 56 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
frigate/motion.py
View File

@@ -5,7 +5,7 @@ from frigate.config import MotionConfig
class MotionDetector:
def __init__(self, frame_shape, config: MotionConfig):
def __init__(self, frame_shape, config: MotionConfig, improve_contrast_enabled):
self.config = config
self.frame_shape = frame_shape
self.resize_factor = frame_shape[0] / config.frame_height
@@ -24,6 +24,7 @@ class MotionDetector:
)
self.mask = np.where(resized_mask == [0])
self.save_images = False
self.improve_contrast = improve_contrast_enabled
def detect(self, frame):
motion_boxes = []
@@ -38,14 +39,15 @@ class MotionDetector:
)
# Improve contrast
minval = np.percentile(resized_frame, 4)
maxval = np.percentile(resized_frame, 96)
# don't adjust if the image is a single color
if minval < maxval:
resized_frame = np.clip(resized_frame, minval, maxval)
resized_frame = (
((resized_frame - minval) / (maxval - minval)) * 255
).astype(np.uint8)
if self.improve_contrast.value:
minval = np.percentile(resized_frame, 4)
maxval = np.percentile(resized_frame, 96)
# don't adjust if the image is a single color
if minval < maxval:
resized_frame = np.clip(resized_frame, minval, maxval)
resized_frame = (
((resized_frame - minval) / (maxval - minval)) * 255
).astype(np.uint8)
# mask frame
resized_frame[self.mask] = [255]