fix: adress review comments about ensure_channel_first

Kirscher · Kirscher · commit f8810adfd5fe · 2025-05-16T13:02:18.000Z
diff --git a/modules/dynunet_pipeline/transforms.py b/modules/dynunet_pipeline/transforms.py
@@ -42,7 +42,7 @@ def get_task_transforms(mode, task_id, pos_sample_num, neg_sample_num, num_sampl
         keys = ["image"]
 
     load_transforms = [
-        LoadImaged(keys=keys, image_only=False, ensure_channel_first=True),
+        LoadImaged(keys=keys, image_only=False),
         EnsureChannelFirstd(keys=keys),
     ]
     # 2. sampling

Original file line number	Diff line number	Diff line change
`@@ -42,7 +42,7 @@ def get_task_transforms(mode, task_id, pos_sample_num, neg_sample_num, num_sampl`
`42`	`42`	`keys = ["image"]`
`43`	`43`
`44`	`44`	`load_transforms = [`
`45`		`- LoadImaged(keys=keys, image_only=False, ensure_channel_first=True),`
	`45`	`+ LoadImaged(keys=keys, image_only=False),`
`46`	`46`	`EnsureChannelFirstd(keys=keys),`
`47`	`47`	`]`
`48`	`48`	`# 2. sampling`