From 94b07dd79ec066b39d49cdbd7045061c1e40363d Mon Sep 17 00:00:00 2001 From: Miro Astore Date: Tue, 20 Aug 2024 10:31:29 -0400 Subject: [PATCH] element wise volume alignment --- src/cryo_challenge/_preprocessing/align_utils.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/cryo_challenge/_preprocessing/align_utils.py b/src/cryo_challenge/_preprocessing/align_utils.py index 4a976b5..d2a7784 100644 --- a/src/cryo_challenge/_preprocessing/align_utils.py +++ b/src/cryo_challenge/_preprocessing/align_utils.py @@ -135,11 +135,10 @@ def align_submission( volumes (torch.Tensor): aligned submission volumes """ for i in range(len(volumes)): - print('aligning ' + str(i) + 'th volume' ) - obj_vol = volumes[i].numpy().astype(np.float32) + obj_vol = volumes[i].numpy().astype(np.float32).copy() obj_vol = Volume(obj_vol / obj_vol.sum()) - ref_vol = Volume(ref_volume / ref_volume.sum()) + ref_vol = Volume(ref_volume.copy() / ref_volume.sum()) _, R_est = align_BO( ref_vol,