fix prior

SyneRBI · Aug 28, 2024 · 3eec68c · 3eec68c
1 parent e114f21
commit 3eec68c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/main_ISTA.py b/main_ISTA.py
@@ -60,6 +60,8 @@ def __init__(self, data: Dataset, num_subsets: int = 7, step_size: float = 0.1,
                                                                     initial_image=data.OSEM_image)
         for f in obj_funs: # add prior to every objective function
             f.set_prior(data.prior)
+        data.prior.set_penalisation_factor(data.prior.get_penalisation_factor() / len(obj_funs))
+        data.prior.set_up(data.OSEM_image)
 
         sampler = Sampler.random_without_replacement(len(obj_funs))
         f = -SGFunction(obj_funs, sampler=sampler)   # negative to turn minimiser into maximiser