Skip to content

Commit 08ccfbd

Browse files
committed
fixing fbp3_tomobar estimator
1 parent d0e79b5 commit 08ccfbd

File tree

1 file changed

+2
-2
lines changed
  • httomo_backends/methods_database/packages/backends/httomolibgpu/supporting_funcs/recon

1 file changed

+2
-2
lines changed

httomo_backends/methods_database/packages/backends/httomolibgpu/supporting_funcs/recon/algorithm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -132,11 +132,11 @@ def _calc_memory_bytes_FBP3d_tomobar(
132132

133133
# 7. astra backprojection will generate an output array
134134
# https://github.com/dkazanc/ToMoBAR/blob/54137829b6326406e09f6ef9c95eb35c213838a7/tomobar/astra_wrappers/astra_base.py#L524
135-
output_dims = _calc_output_dim_FBP3d_tomobar(non_slice_dims_shape, **kwargs)
135+
output_dims = _calc_output_dim_FBP3d_tomobar((angles_tot, det_width), **kwargs)
136136
recon_output_size = np.prod(output_dims) * np.float32().itemsize
137137

138138
# 7. astra backprojection makes a copy of the input
139-
astra_input_slice_size = np.prod(non_slice_dims_shape) * np.float32().itemsize
139+
astra_input_slice_size = (angles_tot * det_width) * np.float32().itemsize
140140

141141
## now we calculate back projection memory (2 copies of the input + reconstruction output)
142142
projection_mem_size = (

0 commit comments

Comments
 (0)