feat: use camera whitebalance instead of calculating a new whitebalan…

…ce. camera whitebalance fixes the blue shift.
UCSD-E4E · Nov 14, 2024 · c820448 · c820448
1 parent 38dc93c
commit c820448
Show file tree

Hide file tree

Showing 4 changed files with 196 additions and 64 deletions.
diff --git a/experiments/color_correction.ipynb b/experiments/color_correction.ipynb
diff --git a/experiments/raw_process.ipynb b/experiments/raw_process.ipynb
diff --git a/experiments/sam_segmentation.ipynb b/experiments/sam_segmentation.ipynb
diff --git a/pyfishsensedev/image/image_processors/raw_processor.py b/pyfishsensedev/image/image_processors/raw_processor.py
@@ -28,7 +28,9 @@ def __next__(self) -> np.ndarray:
         self.__has_iterated = True
 
         with rawpy.imread(self.file.as_posix()) as raw:
-            img = raw.postprocess(gamma=(1, 1), no_auto_bright=True, output_bps=16)
+            img = raw.postprocess(
+                gamma=(1, 1), no_auto_bright=True, use_camera_wb=True, output_bps=16
+            )
             img = skimage.exposure.adjust_gamma(img, gamma=self.__gamma)
             img = skimage.exposure.equalize_adapthist(img)