summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorEdoardo Pasca <edo.paskino@gmail.com>2017-10-31 11:46:58 +0000
committerEdoardo Pasca <edo.paskino@gmail.com>2018-01-19 14:26:06 +0000
commit0b8e774ee12b9d3bbdb644c5046eff490269dc85 (patch)
treee95536e12fa2c08a6981eddeae8f1e3caa76542a
parenta5ee66a4aee472ab72d204783b5e3da4b4f65beb (diff)
downloadregularization-0b8e774ee12b9d3bbdb644c5046eff490269dc85.tar.gz
regularization-0b8e774ee12b9d3bbdb644c5046eff490269dc85.tar.bz2
regularization-0b8e774ee12b9d3bbdb644c5046eff490269dc85.tar.xz
regularization-0b8e774ee12b9d3bbdb644c5046eff490269dc85.zip
Allows regularizer to output simply an image
-rw-r--r--src/Python/ccpi/reconstruction/FISTAReconstructor.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/src/Python/ccpi/reconstruction/FISTAReconstructor.py b/src/Python/ccpi/reconstruction/FISTAReconstructor.py
index 0a66e67..b8e1027 100644
--- a/src/Python/ccpi/reconstruction/FISTAReconstructor.py
+++ b/src/Python/ccpi/reconstruction/FISTAReconstructor.py
@@ -597,7 +597,7 @@ class FISTAReconstructor():
#astra.matlab.data3d('delete', sino_id)
- def regularize(self, X):
+ def regularize(self, X , output_all=False):
print ("FISTA Reconstructor: regularize")
regularizer = self.getParameter('regularizer')
@@ -605,7 +605,8 @@ class FISTAReconstructor():
lc = self.getParameter('Lipschitz_constant')
reg_par = regularizer.getParameter('regularization_parameter') / lc
return regularizer(input=X,
- regularization_parameter = reg_par)
+ regularization_parameter = reg_par,
+ output_all=output_all)
else:
return X