diff options
author | Edoardo Pasca <edo.paskino@gmail.com> | 2017-10-31 16:02:25 +0000 |
---|---|---|
committer | Edoardo Pasca <edo.paskino@gmail.com> | 2018-01-19 14:26:06 +0000 |
commit | 92b18573c8f58ef8ddc6c5f50bfa5adc96501e30 (patch) | |
tree | 0ea724ca914d69ff11389435b253fb7d773876c1 /src | |
parent | ea27a1f3fa285a621b5e327a5a36aef9d5a5018a (diff) | |
download | regularization-92b18573c8f58ef8ddc6c5f50bfa5adc96501e30.tar.gz regularization-92b18573c8f58ef8ddc6c5f50bfa5adc96501e30.tar.bz2 regularization-92b18573c8f58ef8ddc6c5f50bfa5adc96501e30.tar.xz regularization-92b18573c8f58ef8ddc6c5f50bfa5adc96501e30.zip |
reconstructor class working with Ordered Subset
Diffstat (limited to 'src')
-rw-r--r-- | src/Python/ccpi/reconstruction/FISTAReconstructor.py | 23 |
1 files changed, 14 insertions, 9 deletions
diff --git a/src/Python/ccpi/reconstruction/FISTAReconstructor.py b/src/Python/ccpi/reconstruction/FISTAReconstructor.py index 4f1709c..b549755 100644 --- a/src/Python/ccpi/reconstruction/FISTAReconstructor.py +++ b/src/Python/ccpi/reconstruction/FISTAReconstructor.py @@ -431,7 +431,7 @@ class FISTAReconstructor(): # prepareForIteration def iterate (self, Xin=None): - if self.getParameter('subset') == 0: + if self.getParameter('subsets') == 0: return self.iterateStandard(Xin) else: return self.iterateOrderedSubsets(Xin) @@ -608,10 +608,7 @@ class FISTAReconstructor(): regularizer = self.getParameter('regularizer') if regularizer is not None: - lc = self.getParameter('Lipschitz_constant') - reg_par = regularizer.getParameter('regularization_parameter') / lc return regularizer(input=X, - regularization_parameter = reg_par, output_all=output_all) else: return X @@ -642,7 +639,7 @@ class FISTAReconstructor(): print (string.format(i,Resid_error[i], self.objective[i])) return (X , X_t, t) - def iterateOS(self, Xin=None): + def iterateOrderedSubsets(self, Xin=None): print ("FISTA Reconstructor: Ordered Subsets iterate") if Xin is None: @@ -685,7 +682,7 @@ class FISTAReconstructor(): print ("starting iterations") ## % Outer FISTA iterations loop - for i in range(fistaRecon.getParameter('number_of_iterations')): + for i in range(self.getParameter('number_of_iterations')): # With OS approach it becomes trickier to correlate independent # subsets, hence additional work is required one solution is to # work with a full sinogram at times @@ -755,8 +752,11 @@ class FISTAReconstructor(): if lambdaR_L1 > 0 : ## RING REMOVAL print ("ring removal") - residualSub = self.ringRemovalOrderedSubsets(sino_updt_Sub, - sino_updt_FULL) + residualSub = \ + self.ringRemovalOrderedSubsets(ss, + counterInd, + sino_updt_Sub, + sino_updt_FULL) else: #PWLS model print ("PWLS model") @@ -797,7 +797,8 @@ class FISTAReconstructor(): return X - def ringRemovalOrderedSubsets(self, sino_updt_Sub, sino_updt_FULL): + def ringRemovalOrderedSubsets(self, ss,counterInd, + sino_updt_Sub, sino_updt_FULL): residual = self.residual r_x = self.r_x weights , alpha_ring , sino = \ @@ -805,6 +806,10 @@ class FISTAReconstructor(): numProjSub = self.getParameter('os_bins')[ss] CurrSubIndices = self.getParameter('os_indices')\ [counterInd:counterInd+numProjSub] + + shape = list(numpy.shape(self.getParameter('input_sinogram'))) + shape[1] = numProjSub + residualSub = numpy.zeros(shape) for kkk in range(numProjSub): |