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

use clahe for contrast improvement (#6835)

Browse Source 
* use clahe for contrast improvement

* update tests
This commit is contained in:
Blake Blackshear
2023-06-18 11:56:41 -05:00
committed by GitHub
parent 09cc4251f0
commit 7c1568fcb9
3 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
frigate/motion/improved_motion.py
View File

@@ -38,6 +38,7 @@ class ImprovedMotionDetector(MotionDetector):
self.improve_contrast = improve_contrast
self.threshold = threshold
self.contour_area = contour_area
self.clahe = cv2.createCLAHE(clipLimit=2.0, tileGridSize=(8, 8))
def detect(self, frame):
motion_boxes = []
@@ -55,7 +56,7 @@ class ImprovedMotionDetector(MotionDetector):
# Improve contrast
if self.improve_contrast.value:
resized_frame = cv2.equalizeHist(resized_frame)
resized_frame = self.clahe.apply(resized_frame)
# mask frame
resized_frame[self.mask] = [255]