use OSEM image to determine FOV mask

SyneRBI · Aug 30, 2024 · 7f7c288 · 7f7c288
1 parent 08f9808
commit 7f7c288
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/main_SVRG.py b/main_SVRG.py
@@ -134,7 +134,8 @@ def __init__(
             self._adjoint_ones += self._obj_funs[i].get_subset_sensitivity(0)
 
         self._fov_mask = self.x.get_uniform_copy(0)
-        tmp = 1.0 * (self._adjoint_ones.as_array() > 0)
+        # tmp = 1.0 * (self._adjoint_ones.as_array() > 0)
+        tmp = 1.0 * (data.OSEM_image.as_array() > 0)
         self._fov_mask.fill(tmp)
 
         # add a small number in the adjoint ones outside the FOV to avoid NaN in division

diff --git a/test_rdp.py b/test_rdp.py
@@ -386,7 +386,7 @@ def get_image(fname):
         print(srcdir)
         data = get_data(srcdir=srcdir, outdir=outdir)
 
-        print(data.acquired_data.shape)
+        np.save(f"{ds}.npy", data.OSEM_image.as_array())
 
 #        if image_type == "osem":
 #            x = data.OSEM_image