Enforce saturation limit in detector.adc

andykee · Nov 26, 2024 · 654ac74 · 654ac74
1 parent 4cd160a
commit 654ac74
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lentil/detector.py b/lentil/detector.py
@@ -336,6 +336,8 @@ def adc(img, gain, saturation_capacity=None, warn_saturate=False, dtype=None):
     if dtype is not None:
         img = img.astype(dtype)
 
+    img[img > saturation_capacity - 1] = saturation_capacity - 1
+
     return img