Temp disable channel first

Author: C-Achard

napari_cellseg3d/code_models/worker_training.py

@@ -201,11 +201,11 @@ def get_patch_dataset(self, train_transforms):
         patch_func = Compose(
             [
                 LoadImaged(keys=["image"], image_only=True),
-                EnsureChannelFirstd(
-                    keys=["image"],
-                    channel_dim="no_channel",
-                    strict_check=False,
-                ),
+                # EnsureChannelFirstd(
+                #     keys=["image"],
+                #     channel_dim="no_channel",
+                #     strict_check=False,
+                # ),
                 RandSpatialCropSamplesd(
                     keys=["image"],
                     roi_size=(
@@ -287,11 +287,11 @@ def get_dataset(self, train_transforms):
         load_single_images = Compose(
             [
                 LoadImaged(keys=["image"]),
-                EnsureChannelFirstd(
-                    keys=["image"],
-                    channel_dim="no_channel",
-                    strict_check=False,
-                ),
+                # EnsureChannelFirstd(
+                #     keys=["image"],
+                #     channel_dim="no_channel",
+                #     strict_check=False,
+                # ),
                 Orientationd(keys=["image"], axcodes="PLI"),
                 SpatialPadd(
                     keys=["image"],