diff options
author | Edoardo Pasca <edo.paskino@gmail.com> | 2019-05-10 11:51:05 +0100 |
---|---|---|
committer | Edoardo Pasca <edo.paskino@gmail.com> | 2019-05-10 11:51:05 +0100 |
commit | 52b7eff6f554c74d9e7e3f0d38a126bef90ee9f8 (patch) | |
tree | 5b5eabad6cf12586d43d28b22eedda0a3882b936 /Wrappers/Python | |
parent | 4b68c798e24543639994c3105c720527cdd23df8 (diff) | |
download | framework-52b7eff6f554c74d9e7e3f0d38a126bef90ee9f8.tar.gz framework-52b7eff6f554c74d9e7e3f0d38a126bef90ee9f8.tar.bz2 framework-52b7eff6f554c74d9e7e3f0d38a126bef90ee9f8.tar.xz framework-52b7eff6f554c74d9e7e3f0d38a126bef90ee9f8.zip |
uses Gradient.CORRELATION_SPACE variable, reduce iterations
Diffstat (limited to 'Wrappers/Python')
-rw-r--r-- | Wrappers/Python/demos/PDHG_examples/PDHG_TV_Denoising_Gaussian.py | 6 |
1 files 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() |