浏览代码

avoid divide by zero

Blake Blackshear 3 年之前
父节点
当前提交
543a8a1712
共有 1 个文件被更改,包括 6 次插入4 次删除
  1. 6 4
      frigate/motion.py

+ 6 - 4
frigate/motion.py

@@ -40,10 +40,12 @@ class MotionDetector:
         # Improve contrast
         minval = np.percentile(resized_frame, 4)
         maxval = np.percentile(resized_frame, 96)
-        resized_frame = np.clip(resized_frame, minval, maxval)
-        resized_frame = (((resized_frame - minval) / (maxval - minval)) * 255).astype(
-            np.uint8
-        )
+        # 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]