From ebbf875a6baeb0e0a77f2d90094538314c4fe91c Mon Sep 17 00:00:00 2001 From: Edoardo Pasca Date: Thu, 22 Mar 2018 14:33:57 +0000 Subject: bugfix: use get_acquisition_data --- Wrappers/Python/wip/test_reader_reconstr.py | 18 +++--------------- 1 file changed, 3 insertions(+), 15 deletions(-) (limited to 'Wrappers/Python') diff --git a/Wrappers/Python/wip/test_reader_reconstr.py b/Wrappers/Python/wip/test_reader_reconstr.py index 79afd61..325437c 100755 --- a/Wrappers/Python/wip/test_reader_reconstr.py +++ b/Wrappers/Python/wip/test_reader_reconstr.py @@ -39,10 +39,9 @@ print (dims) flat = avg_img(reader.load_flat()) dark = avg_img(reader.load_dark()) -data = reader.getAcquisitionData() norm = Normalizer(flat_field=flat, dark_field=dark) -norm.set_input(reader.getAcquisitionData()) +norm.set_input(reader.get_acquisition_data()) cor = CenterOfRotationFinder() cor.set_input(norm.get_output()) @@ -78,7 +77,7 @@ x_init = ImageData(geometry=vg, dimension_labels=['horizontal_x','horizontal_y', #%% print ("run FISTA") # Run FISTA for least squares without regularization -opt = {'tol': 1e-4, 'iter': 500} +opt = {'tol': 1e-4, 'iter': 10} x_fista0, it0, timing0, criter0 = FISTA(x_init, f, None, opt=opt) pickle.dump(x_fista0, open("fista0.pkl", "wb")) @@ -139,8 +138,7 @@ plt.title('CGLS criterion') #plt.show() -clims = (0,1) -cols = 5 +cols = 4 rows = 1 current = 1 fig = plt.figure() @@ -166,16 +164,6 @@ a=fig.add_subplot(rows,cols,current) a.set_title('CGLS') imgplot = plt.imshow(x_CGLS.subset(horizontal_x=80).as_array()) -current = current + 1 -a=fig.add_subplot(rows,cols,current) -a.set_title('criteria') -imgplot = plt.loglog(criter0 , label='FISTA0') -imgplot = plt.loglog(criter1 , label='FISTA1') -imgplot = plt.loglog(criter_fbpd1, label='FBPD1') -imgplot = plt.loglog(criter_CGLS, label='CGLS') -#imgplot = plt.loglog(criter_fbpdtv, label='FBPD TV') -a.legend(loc='right') - plt.show() -- cgit v1.2.3