Issue410: Added groundtruth comparion flags to ./python/TIGRE/algorithms

Python/tigre/algorithms/iterative_recon_alg.py

@@ -92,6 +92,12 @@ class IterativeReconAlg(object):
             OS_SART_TV
             FISTA
 
+
+    :keyword groundtruth: ()
+        groundtruth image for comparison with the reconstruction.
+        Default is None
+
+
     Usage
     --------
     >>> import numpy as np
@@ -133,6 +139,9 @@ def __init__(self, proj, geo, angles, niter, **kwargs):
         self.angles = angles
         self.geo = geo
         self.niter = niter
+
+        # adding groundtruth comparison
+        self.groundtruth = None if "groundtruth" not in kwargs else kwargs["groundtruth"]
 
         self.geo.check_geo(angles)
 
@@ -203,6 +212,9 @@ def __init__(self, proj, geo, angles, niter, **kwargs):
             self.Quameasopts = (
                 [self.Quameasopts] if isinstance(self.Quameasopts, str) else self.Quameasopts
             )
+            # appending error_norm is groundtruth is Null
+            if self.groundtruth is not None:
+                self.Quameasopts.append('error_norm')
             setattr(self, "lq", np.zeros([len(self.Quameasopts), niter]))  # quameasoptslist
         else:
             setattr(self, "lq", np.zeros([0, niter]))  # quameasoptslist
@@ -311,7 +323,8 @@ def run_main_iter(self):
         Quameasopts = self.Quameasopts
 
         for i in range(self.niter):
-
+            # init Quameasopts
+            Quameasopts = self.Quameasopts
             res_prev = None
             if Quameasopts is not None:
                 res_prev = copy.deepcopy(self.res)
@@ -357,7 +370,10 @@ def minimizeAwTV(self, res_prev, dtvg):
 
     def error_measurement(self, res_prev, iter):
         if self.Quameasopts is not None:
-            self.lq[:, iter] = MQ(self.res, res_prev, self.Quameasopts)
+            if self.groundtruth is not None:
+                self.lq[:, iter] = MQ(self.res, self.groundtruth, self.Quameasopts)
+            else:
+                self.lq[:, iter] = MQ(self.res, res_prev, self.Quameasopts)
         if self.computel2:
             # compute l2 borm for b-Ax
             errornow = im3DNORM(
@@ -481,3 +497,5 @@ def iterativereconalg(proj, geo, angles, niter, **kwargs):
 
 
 iterativereconalg = decorator(IterativeReconAlg)
+
+# have to find a usecase to groundtruth comparing with matlab

Python/tigre/algorithms/single_pass_algorithms.py

@@ -70,9 +70,10 @@ def FDK(proj, geo, angles, **kwargs):
                         PYTHON : Reuben Lindroos
     """
     verbose = kwargs["verbose"] if "verbose" in kwargs else False
-
     gpuids = kwargs["gpuids"] if "gpuids" in kwargs else None
     dowang = kwargs["dowang"] if "dowang" in kwargs else False
+    # adding groundtruth comparison
+    groundtruth = None if "groundtruth" not in kwargs else kwargs["groundtruth"]
 
     def zeropadding(proj, geo):
         zgeo = copy.deepcopy(geo)
@@ -192,4 +193,6 @@ def fbp(proj, geo, angles, **kwargs):  # noqa: D103
     gpuids = kwargs["gpuids"] if "gpuids" in kwargs else None
     proj_filt = filtering(copy.deepcopy(proj), geox,
                           angles, parker=False, verbose=verbose)
+    # adding groundtruth comparison 
+    groundtruth = None if "groundtruth" not in kwargs else kwargs["groundtruth"]
     return Atb(proj_filt, geo, angles, gpuids=gpuids) * geo.DSO / geo.DSD