Enable ChannelFirst again

Author: C-Achard

napari_cellseg3d/dev_scripts/colab_training.py

@@ -9,13 +9,11 @@
 # MONAI
 from monai.metrics import DiceMetric
 from monai.transforms import (
-    AddChanneld,
     Compose,
     EnsureChannelFirstd,
     EnsureTyped,
     LoadImaged,
     Orientationd,
-    SpatialPadd,
 )
 
 # local
@@ -128,12 +126,11 @@ def get_dataset(self, train_transforms):
         load_single_images = Compose(
             [
                 LoadImaged(keys=["image"]),
-                # EnsureChannelFirstd(
-                #     keys=["image"],
-                #     channel_dim="no_channel",
-                #     strict_check=False,
-                # ),
-                AddChanneld(keys=["image"]),
+                EnsureChannelFirstd(
+                    keys=["image"],
+                    channel_dim="no_channel",
+                    strict_check=False,
+                ),
                 Orientationd(keys=["image"], axcodes="PLI"),
                 # SpatialPadd(
                 #     keys=["image"],