Update background.py

SINTEF · Sep 25, 2024 · 5698e5f · 5698e5f
1 parent a7bb309
commit 5698e5f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyopia/background.py b/pyopia/background.py
@@ -172,8 +172,8 @@ def correct_im_accurate_divide(imbg, imraw):
         corrected image, same type as input
     '''
 
-    zero_mask = imbg == 0   # a mask to find pixels with zero value
-    imbg[zero_mask] = 1 / 255   # change the zero_value pixels to prevent RuntimeWarning: 'divide by zero'
+    for i in range(imbg.shape[2]):
+        imbg[:, :, i][imbg[:, :, i] == 0] = 1 / 255   # change the zero_value pixels to prevent RuntimeWarning: 'divide by zero'
 
     im_corrected = imraw / imbg