Add black level assertion.

LCAV · Nov 29, 2023 · 1b21fdb · 1b21fdb
1 parent 8e3747b
commit 1b21fdb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lensless/utils/image.py b/lensless/utils/image.py
@@ -265,6 +265,9 @@ def bayer2rgb_cc(
         dtype = np.uint16
     else:
         dtype = np.uint8
+    assert (
+        black_level < 2**nbits - 1
+    ), f"Black level ({black_level}) too high for bit depth ({nbits})."
 
     # demosaic Bayer data
     img = cv2.cvtColor(img, cv2.COLOR_BayerRG2RGB)