diff options
| author | algol <dkazanc@hotmail.com> | 2018-02-22 12:34:39 +0000 |
|---|---|---|
| committer | algol <dkazanc@hotmail.com> | 2018-02-22 12:34:39 +0000 |
| commit | d4d12945f5396b10259fdeeb2b09f99b0e2c6afd (patch) | |
| tree | 378fcd77b72b7e433db2e2b2e121d34d479e988d /Wrappers/Python/demo | |
| parent | ce313c7dcf45edaf4c824690d8caa7df5df90120 (diff) | |
| download | regularization-d4d12945f5396b10259fdeeb2b09f99b0e2c6afd.tar.gz regularization-d4d12945f5396b10259fdeeb2b09f99b0e2c6afd.tar.bz2 regularization-d4d12945f5396b10259fdeeb2b09f99b0e2c6afd.tar.xz regularization-d4d12945f5396b10259fdeeb2b09f99b0e2c6afd.zip | |
GPU test fixed, cpu vs gpu test added for ROF-TV
Diffstat (limited to 'Wrappers/Python/demo')
| -rw-r--r-- | Wrappers/Python/demo/test_cpu_regularizers.py | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/Wrappers/Python/demo/test_cpu_regularizers.py b/Wrappers/Python/demo/test_cpu_regularizers.py index 5908c3c..d147b85 100644 --- a/Wrappers/Python/demo/test_cpu_regularizers.py +++ b/Wrappers/Python/demo/test_cpu_regularizers.py @@ -284,9 +284,9 @@ start_time = timeit.default_timer() pars = {'algorithm': ROF_TV , \ 'input' : u0,\ - 'regularization_parameter':1,\ - 'marching_step': 0.003,\ - 'number_of_iterations': 300 + 'regularization_parameter':25,\ + 'marching_step': 0.001,\ + 'number_of_iterations': 350 } rof = ROF_TV(pars['input'], pars['number_of_iterations'], @@ -307,9 +307,7 @@ props = dict(boxstyle='round', facecolor='wheat', alpha=0.5) # place a text box in upper left in axes coords a.text(0.05, 0.95, txtstr, transform=a.transAxes, fontsize=14, verticalalignment='top', bbox=props) -imgplot = plt.imshow(tgv, cmap="gray") - - +imgplot = plt.imshow(rof, cmap="gray") plt.show() |