From f5aa0dd2976df492374b35d9f6bfcc0933803ebb Mon Sep 17 00:00:00 2001
From: "Jakob Jorgensen, WS at HMXIF" <jakob.jorgensen@manchester.ac.uk>
Date: Sat, 12 May 2018 21:07:29 +0100
Subject: Add SB in final display of all.

---
 Wrappers/Python/wip/demo_compare_RGLTK_TV_denoising.py | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

(limited to 'Wrappers/Python/wip')

diff --git a/Wrappers/Python/wip/demo_compare_RGLTK_TV_denoising.py b/Wrappers/Python/wip/demo_compare_RGLTK_TV_denoising.py
index 2bf1286..fd992de 100644
--- a/Wrappers/Python/wip/demo_compare_RGLTK_TV_denoising.py
+++ b/Wrappers/Python/wip/demo_compare_RGLTK_TV_denoising.py
@@ -149,7 +149,7 @@ print(EnergytotalSB)
 # Compare all reconstruction
 clims = (-0.2,1.2)
 dlims = (-0.2,0.2)
-cols = 3
+cols = 4
 rows = 2
 current = 1
 
@@ -171,6 +171,12 @@ a.set_title('FGP')
 imgplot = plt.imshow(xtv_fgp.as_array(),vmin=clims[0],vmax=clims[1])
 plt.axis('off')
 
+current = current + 1
+a=fig.add_subplot(rows,cols,current)
+a.set_title('SB')
+imgplot = plt.imshow(xtv_sb.as_array(),vmin=clims[0],vmax=clims[1])
+plt.axis('off')
+
 current = current + 1
 a=fig.add_subplot(rows,cols,current)
 a.set_title('FBPD - CVX')
@@ -179,12 +185,18 @@ plt.axis('off')
 
 current = current + 1
 a=fig.add_subplot(rows,cols,current)
-a.set_title('ROF - TV')
+a.set_title('ROF - CVX')
 imgplot = plt.imshow(xtv_rof.as_array()-xtv_denoise.value,vmin=dlims[0],vmax=dlims[1])
 plt.axis('off')
 
 current = current + 1
 a=fig.add_subplot(rows,cols,current)
-a.set_title('FGP - TV')
+a.set_title('FGP - CVX')
 imgplot = plt.imshow(xtv_fgp.as_array()-xtv_denoise.value,vmin=dlims[0],vmax=dlims[1])
 plt.axis('off')
+
+current = current + 1
+a=fig.add_subplot(rows,cols,current)
+a.set_title('SB - CVX')
+imgplot = plt.imshow(xtv_sb.as_array()-xtv_denoise.value,vmin=dlims[0],vmax=dlims[1])
+plt.axis('off')
-- 
cgit v1.2.3