summaryrefslogtreecommitdiffstats
path: root/Wrappers/Python
diff options
context:
space:
mode:
authorEdoardo Pasca <edo.paskino@gmail.com>2019-05-10 11:51:05 +0100
committerEdoardo Pasca <edo.paskino@gmail.com>2019-05-10 11:51:05 +0100
commit52b7eff6f554c74d9e7e3f0d38a126bef90ee9f8 (patch)
tree5b5eabad6cf12586d43d28b22eedda0a3882b936 /Wrappers/Python
parent4b68c798e24543639994c3105c720527cdd23df8 (diff)
downloadframework-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.py6
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()