summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
authorEdoardo Pasca <edo.paskino@gmail.com>2017-10-31 11:46:58 +0000
committerEdoardo Pasca <edo.paskino@gmail.com>2017-10-31 11:46:58 +0000
commit4bbfd47b88b34b7a7ba1545f5f47de1fc4d15cc5 (patch)
tree44dba6affc1f6f8a918e4574e5adb56dffed4ec9 /src
parent8bc351e5cba87769c3ed0e0c872e823f7f5943fb (diff)
downloadregularization-4bbfd47b88b34b7a7ba1545f5f47de1fc4d15cc5.tar.gz
regularization-4bbfd47b88b34b7a7ba1545f5f47de1fc4d15cc5.tar.bz2
regularization-4bbfd47b88b34b7a7ba1545f5f47de1fc4d15cc5.tar.xz
regularization-4bbfd47b88b34b7a7ba1545f5f47de1fc4d15cc5.zip
Allows regularizer to output simply an image
Diffstat (limited to 'src')
-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