diff options
| -rwxr-xr-x | Wrappers/Python/wip/test_reader_reconstr.py | 18 |
1 files changed, 3 insertions, 15 deletions
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()
|