diff --git a/flextomo/projector.py b/flextomo/projector.py index 030caa9..aff1d29 100644 --- a/flextomo/projector.py +++ b/flextomo/projector.py @@ -117,9 +117,10 @@ def SIRT( projections, volume, geometry, iterations): # Stop progress bar _pbar_close_(pbar) - - if rnorms: - display.plot2d(rnorms, semilogy = True, title = 'Resudual L2') + + if rnorms: + display.plot2d(rnorms, semilogy = True, title = 'Resudual L2') + return rnorms def PWLS(projections, volume, geometry, iterations): """ @@ -892,5 +893,5 @@ def _subset_generator_(projections, volume, geometry, copy = False): # Initialize ASTRA geometries: proj_geom = geometry[jj].astra_projection_geom(projections[jj].shape, index = sl) vol_geom = geometry[jj].astra_volume_geom(volume.shape) - - yield subset, proj_geom, vol_geom \ No newline at end of file + + yield subset, proj_geom, vol_geom