From 52b7eff6f554c74d9e7e3f0d38a126bef90ee9f8 Mon Sep 17 00:00:00 2001 From: Edoardo Pasca Date: Fri, 10 May 2019 11:51:05 +0100 Subject: uses Gradient.CORRELATION_SPACE variable, reduce iterations --- Wrappers/Python/demos/PDHG_examples/PDHG_TV_Denoising_Gaussian.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/Wrappers/Python/demos/PDHG_examples/PDHG_TV_Denoising_Gaussian.py b/Wrappers/Python/demos/PDHG_examples/PDHG_TV_Denoising_Gaussian.py index 7966ded..ab4e89c 100644 --- a/Wrappers/Python/demos/PDHG_examples/PDHG_TV_Denoising_Gaussian.py +++ b/Wrappers/Python/demos/PDHG_examples/PDHG_TV_Denoising_Gaussian.py @@ -93,14 +93,14 @@ plt.colorbar() plt.show() # Regularisation Parameter -alpha = 2. +alpha = .1 method = '0' if method == '0': # Create operators - op1 = Gradient(ig, correlation='SpaceChannels') + op1 = Gradient(ig, correlation=Gradient.CORRELATION_SPACE) op2 = Identity(ig, ag) # Create BlockOperator @@ -131,7 +131,7 @@ tau = 1/(sigma*normK**2) pdhg = PDHG(f=f,g=g,operator=operator, tau=tau, sigma=sigma) pdhg.max_iteration = 10000 pdhg.update_objective_interval = 100 -pdhg.run(1000, verbose=True) +pdhg.run(200, verbose=True) # Show Results plt.figure() -- cgit v1.2.3