From 1062346b9f892b7a90ccfb8406bd61fdc44e8e82 Mon Sep 17 00:00:00 2001
From: Edoardo Pasca <edo.paskino@gmail.com>
Date: Thu, 2 Nov 2017 18:03:41 +0000
Subject: added demos
---
src/Python/demo/demo_dendrites.py | 36 +++++++++++++++++++++++++++++++++---
1 file changed, 33 insertions(+), 3 deletions(-)
(limited to 'src/Python/demo')
diff --git a/src/Python/demo/demo_dendrites.py b/src/Python/demo/demo_dendrites.py
index 528702c..f5dc845 100644
--- a/src/Python/demo/demo_dendrites.py
+++ b/src/Python/demo/demo_dendrites.py
@@ -1,3 +1,4 @@
+
# -*- coding: utf-8 -*-
"""
Created on Wed Aug 23 16:34:49 2017
@@ -86,14 +87,14 @@ fistaRecon = FISTAReconstructor(proj_geom,
subsets = 8)
print("Reconstruction using FISTA-OS-PWLS without regularization...")
-fistaRecon.setParameter(number_of_iterations = 18)
+fistaRecon.setParameter(number_of_iterations = 5)
### adjust the regularization parameter
##lc = fistaRecon.getParameter('Lipschitz_constant')
##fistaRecon.getParameter('regularizer')\
## .setParameter(regularization_parameter=5e6/lc)
fistaRecon.use_device = True
-if False:
+if True:
fistaRecon.prepareForIteration()
X = fistaRecon.iterate(numpy.load("../test/X.npy"))
numpy.save("FISTA-OS-PWLS.npy",X)
@@ -123,7 +124,7 @@ if False:
X = fistaRecon.iterate(numpy.load("../test/X.npy"))
numpy.save("FISTA-OS-GH-TV.npy",X)
-if True:
+if False:
# adjust the regularization parameter
lc = fistaRecon.getParameter('Lipschitz_constant')
regul.setParameter(
@@ -136,3 +137,32 @@ if True:
X = fistaRecon.iterate(numpy.load("../test/X.npy"))
numpy.save("FISTA-OS-GH-TGV.npy",X)
+if False:
+ # adjust the regularization parameter
+ lc = fistaRecon.getParameter('Lipschitz_constant')
+ regul.setParameter(
+ algorithm=Regularizer.Algorithm.PatchBased_Regul,
+ regularization_parameter=3/lc,
+ searching_window_ratio=3,
+ similarity_window_ratio=1,
+ PB_filtering_parameter=0.04
+
+ )
+ fistaRecon.setParameter(regularizer=regul)
+ fistaRecon.setParameter(ring_lambda_R_L1=0.002, ring_alpha=21)
+ fistaRecon.prepareForIteration()
+ X = fistaRecon.iterate(numpy.load("../test/X.npy"))
+ numpy.save("FISTA-OS-CPU_PB.npy",X)
+
+if False:
+ fistaRecon = FISTAReconstructor(proj_geom,
+ vol_geom,
+ Sino3D ,
+ weights=Weights3D,
+ device=astradevice,
+ Lipschitz_constant = 7.6792e8,
+ number_of_iterations=50)
+
+ fistaRecon.prepareForIteration()
+ X = fistaRecon.iterate(numpy.load("../test/X.npy"))
+ numpy.save("FISTA.npy",X)
--
cgit v1.2.3