From 4bbfd47b88b34b7a7ba1545f5f47de1fc4d15cc5 Mon Sep 17 00:00:00 2001 From: Edoardo Pasca Date: Tue, 31 Oct 2017 11:46:58 +0000 Subject: Allows regularizer to output simply an image --- src/Python/ccpi/reconstruction/FISTAReconstructor.py | 5 +++-- 1 file 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 -- cgit v1.2.3