Bug fix: masking non exposure area in RL algorithm

hiyoneda · Jul 10, 2024 · 170c618 · 170c618
1 parent 7b5715c
commit 170c618
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cosipy/image_deconvolution/RichardsonLucy.py b/cosipy/image_deconvolution/RichardsonLucy.py
@@ -126,6 +126,9 @@ def Mstep(self):
         # delta model
         sum_T_product = self.calc_summed_T_product(ratio_list)
         self.delta_model = self.model * (sum_T_product/self.summed_exposure_map - 1)
+
+        if self.mask is not None:
+            self.delta_model = self.delta_model.mask_pixels(self.mask)
 
         # background normalization optimization
         if self.do_bkg_norm_optimization: