Skip to content

Commit 3da3464

Browse files
committed
corrections to logpolar estimator
1 parent f73d053 commit 3da3464

File tree

1 file changed

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

1 file changed

+2
-4
lines changed

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

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,7 @@ def _calc_memory_bytes_LPRec3d_tomobar(
219219
if odd_horiz:
220220
output_dims = tuple(x + 1 for x in output_dims)
221221

222-
in_slice_size = np.prod(non_slice_dims_shape) * dtype.itemsize
222+
in_slice_size = (angles_tot * DetectorsLengthH) * dtype.itemsize
223223
padded_in_slice_size = angles_tot * n * np.float32().itemsize
224224

225225
theta_size = angles_tot * np.float32().itemsize
@@ -352,9 +352,7 @@ def add_to_memory_counters(amount, per_slice: bool):
352352
add_to_memory_counters(circular_mask_size, False)
353353
add_to_memory_counters(after_recon_swapaxis_slice, True)
354354

355-
return (int(tot_memory_bytes * 1.05), fixed_amount + 250 * 1024 * 1024)
356-
# return (tot_memory_bytes, fixed_amount)
357-
355+
return (tot_memory_bytes * 1.05, fixed_amount + 250 * 1024 * 1024)
358356

359357
def _calc_memory_bytes_SIRT3d_tomobar(
360358
non_slice_dims_shape: Tuple[int, int],

0 commit comments

Comments
 (0)