From 95026a236ba8e21bf485080ce9e8eeb0e71ad51a Mon Sep 17 00:00:00 2001
From: Edoardo Pasca <edo.paskino@gmail.com>
Date: Sun, 18 Feb 2018 22:10:53 +0000
Subject: added ROF_TV to CPU demo
---
Wrappers/Python/demo/test_cpu_regularizers.py | 59 +++++++++++++++++++++------
1 file changed, 47 insertions(+), 12 deletions(-)
(limited to 'Wrappers/Python/demo')
diff --git a/Wrappers/Python/demo/test_cpu_regularizers.py b/Wrappers/Python/demo/test_cpu_regularizers.py
index 03d650f..1cac7b9 100644
--- a/Wrappers/Python/demo/test_cpu_regularizers.py
+++ b/Wrappers/Python/demo/test_cpu_regularizers.py
@@ -14,6 +14,7 @@ import timeit
from ccpi.filters.cpu_regularizers_boost import SplitBregman_TV , FGP_TV ,\
LLT_model, PatchBased_Regul ,\
TGV_PD
+from ccpi.filters.cpu_regularizers_cython import ROF_TV
###############################################################################
#https://stackoverflow.com/questions/13875989/comparing-image-in-url-to-image-in-filesystem-in-python/13884956#13884956
@@ -81,7 +82,7 @@ u0 = f(u0).astype('float32')
## plot
fig = plt.figure()
-a=fig.add_subplot(2,3,1)
+a=fig.add_subplot(2,4,1)
a.set_title('noise')
imgplot = plt.imshow(u0,cmap="gray"
)
@@ -114,7 +115,7 @@ txtstr += "%s = %.3fs" % ('elapsed time',timeit.default_timer() - start_time)
print (txtstr)
-a=fig.add_subplot(2,3,2)
+a=fig.add_subplot(2,4,2)
# these are matplotlib.patch.Patch properties
@@ -152,7 +153,7 @@ txtstr += "%s = %.3fs" % ('elapsed time',timeit.default_timer() - start_time)
print (txtstr)
-a=fig.add_subplot(2,3,3)
+a=fig.add_subplot(2,4,3)
# these are matplotlib.patch.Patch properties
props = dict(boxstyle='round', facecolor='wheat', alpha=0.5)
@@ -190,7 +191,7 @@ pars['rmse'] = rms
txtstr = printParametersToString(pars)
txtstr += "%s = %.3fs" % ('elapsed time',timeit.default_timer() - start_time)
print (txtstr)
-a=fig.add_subplot(2,3,4)
+a=fig.add_subplot(2,4,4)
# these are matplotlib.patch.Patch properties
props = dict(boxstyle='round', facecolor='wheat', alpha=0.5)
@@ -230,7 +231,7 @@ txtstr = printParametersToString(pars)
txtstr += "%s = %.3fs" % ('elapsed time',timeit.default_timer() - start_time)
print (txtstr)
-a=fig.add_subplot(2,3,5)
+a=fig.add_subplot(2,4,5)
# these are matplotlib.patch.Patch properties
@@ -268,7 +269,7 @@ pars['rmse'] = rms
txtstr = printParametersToString(pars)
txtstr += "%s = %.3fs" % ('elapsed time',timeit.default_timer() - start_time)
print (txtstr)
-a=fig.add_subplot(2,3,6)
+a=fig.add_subplot(2,4,6)
# these are matplotlib.patch.Patch properties
props = dict(boxstyle='round', facecolor='wheat', alpha=0.5)
@@ -276,6 +277,40 @@ props = dict(boxstyle='round', facecolor='wheat', alpha=0.5)
a.text(0.05, 0.95, txtstr, transform=a.transAxes, fontsize=14,
verticalalignment='top', bbox=props)
imgplot = plt.imshow(tgv, cmap="gray")
+
+# ###################### ROF_TV #########################################
+
+start_time = timeit.default_timer()
+
+pars = {'algorithm': ROF_TV , \
+ 'input' : u0,\
+ 'regularization_parameter':0.07,\
+ 'marching_step': 1.3,\
+ 'number_of_iterations': 550
+ }
+rof = ROF_TV(pars['input'],
+ pars['number_of_iterations'],
+ pars['regularization_parameter'],
+ pars['marching_step']
+ )
+#tgv = out
+rms = rmse(Im, rof)
+pars['rmse'] = rms
+
+txtstr = printParametersToString(pars)
+txtstr += "%s = %.3fs" % ('elapsed time',timeit.default_timer() - start_time)
+print (txtstr)
+a=fig.add_subplot(2,4,7)
+
+# these are matplotlib.patch.Patch properties
+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")
+
+
+
plt.show()
################################################################################
@@ -320,7 +355,7 @@ plt.show()
##imgplot = plt.imshow(Im)
#sliceNo = 32
#
-#a=fig3D.add_subplot(2,3,1)
+#a=fig3D.add_subplot(2,4,1)
#a.set_title('noise')
#imgplot = plt.imshow(u0.T[sliceNo])
#
@@ -346,7 +381,7 @@ plt.show()
#pars = out2[-2]
#reg_output3d.append(out2)
#
-#a=fig3D.add_subplot(2,3,2)
+#a=fig3D.add_subplot(2,4,2)
#
#
#textstr = out2[-1]
@@ -366,7 +401,7 @@ plt.show()
#pars = out2[-2]
#reg_output3d.append(out2)
#
-#a=fig3D.add_subplot(2,3,2)
+#a=fig3D.add_subplot(2,4,2)
#
#
#textstr = out2[-1]
@@ -392,7 +427,7 @@ plt.show()
#pars = out2[-2]
#reg_output3d.append(out2)
#
-#a=fig3D.add_subplot(2,3,2)
+#a=fig3D.add_subplot(2,4,2)
#
#
#textstr = out2[-1]
@@ -418,7 +453,7 @@ plt.show()
#pars = out2[-2]
#reg_output3d.append(out2)
#
-#a=fig3D.add_subplot(2,3,2)
+#a=fig3D.add_subplot(2,4,2)
#
#
#textstr = out2[-1]
@@ -446,7 +481,7 @@ plt.show()
#pars = out2[-2]
#reg_output3d.append(out2)
#
-#a=fig3D.add_subplot(2,3,2)
+#a=fig3D.add_subplot(2,4,2)
#
#
#textstr = out2[-1]
--
cgit v1.2.3