From d19414b27f13b16bbccbfed82e8f94003e9a5ad4 Mon Sep 17 00:00:00 2001 From: Don Yin Date: Wed, 28 Oct 2026 12:02:45 +0000 Subject: [PATCH] Use reoriented image shape for MAISI resize dims Signed-off-by: Don Yin --- .../maisi/scripts/diff_model_create_training_data.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/generation/maisi/scripts/diff_model_create_training_data.py b/generation/maisi/scripts/diff_model_create_training_data.py index 177dfa34c..562ff5151 100644 --- a/generation/maisi/scripts/diff_model_create_training_data.py +++ b/generation/maisi/scripts/diff_model_create_training_data.py @@ -194,12 +194,8 @@ def diff_model_create_training_data( continue filepath = filenames_raw[_iter] - new_dim = tuple( - round_number( - int(plain_transforms({"image": os.path.join(args.data_base_dir, filepath)})["image"].meta["dim"][_i]) - ) - for _i in range(1, 4) - ) + image = plain_transforms({"image": os.path.join(args.data_base_dir, filepath)})["image"] + new_dim = tuple(round_number(dim) for dim in image.shape[1:]) new_transforms = create_transforms(new_dim) process_file(filepath, args, autoencoder, device, plain_transforms, new_transforms, logger)