summaryrefslogtreecommitdiffstats
path: root/Wrappers
diff options
context:
space:
mode:
authorepapoutsellis <epapoutsellis@gmail.com>2019-04-30 14:56:27 +0100
committerepapoutsellis <epapoutsellis@gmail.com>2019-04-30 14:56:27 +0100
commitab97687bd091f523beaeca897a98164318ce3c52 (patch)
tree96d37897fb1b5dcfb38cf9319d1cee8cbd99736e /Wrappers
parentf0e113082137af07d3a184b238f6b50bee395ff2 (diff)
downloadframework-ab97687bd091f523beaeca897a98164318ce3c52.tar.gz
framework-ab97687bd091f523beaeca897a98164318ce3c52.tar.bz2
framework-ab97687bd091f523beaeca897a98164318ce3c52.tar.xz
framework-ab97687bd091f523beaeca897a98164318ce3c52.zip
test IMAT wb recon
Diffstat (limited to 'Wrappers')
-rw-r--r--Wrappers/Python/wip/Demos/IMAT_Reconstruction/TV_WhiteBeam_reconstruction.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/Wrappers/Python/wip/Demos/IMAT_Reconstruction/TV_WhiteBeam_reconstruction.py b/Wrappers/Python/wip/Demos/IMAT_Reconstruction/TV_WhiteBeam_reconstruction.py
index 10d15fa..e67bdb1 100644
--- a/Wrappers/Python/wip/Demos/IMAT_Reconstruction/TV_WhiteBeam_reconstruction.py
+++ b/Wrappers/Python/wip/Demos/IMAT_Reconstruction/TV_WhiteBeam_reconstruction.py
@@ -56,7 +56,7 @@ with open("golden_angles.txt") as f:
angles_string = [line.rstrip() for line in f]
angles = numpy.array(angles_string).astype(float)
agWB = AcquisitionGeometry('parallel', '2D', angles * numpy.pi / 180, pixh)
-op_WB = AstraProjectorSimple(igWB, agWB, 'cpu')
+op_WB = AstraProjectorSimple(igWB, agWB, 'gpu')
sinogram_aqdata = AcquisitionData(sinogram_wb, agWB)
# BackProjection
@@ -71,7 +71,7 @@ plt.show()
#%%
# Regularisation Parameter
-alpha = 10
+alpha = 2000
# Create operators
op1 = Gradient(igWB)
@@ -89,7 +89,7 @@ f = BlockFunction(f1, f2)
g = ZeroFunction()
-diag_precon = True
+diag_precon = False
if diag_precon:
@@ -114,8 +114,8 @@ else:
## Primal & dual stepsizes
-sigma = 0.1
-tau = 1/(sigma*normK**2)
+#sigma = 0.1
+#tau = 1/(sigma*normK**2)
#
#
## Setup and run the PDHG algorithm