Remove strict channel first

Author: C-Achard

napari_cellseg3d/code_models/worker_training.py

@@ -200,7 +200,7 @@ def get_patch_dataset(self, train_transforms):
         patch_func = Compose(
             [
                 LoadImaged(keys=["image"], image_only=True),
-                EnsureChannelFirstd(keys=["image"], channel_dim="no_channel"),
+                EnsureChannelFirstd(keys=["image"], strict_check=False),
                 RandSpatialCropSamplesd(
                     keys=["image"],
                     roi_size=(
@@ -235,7 +235,7 @@ def get_dataset_eval(self, eval_dataset_dict):
             [
                 LoadImaged(keys=["image", "label"]),
                 EnsureChannelFirstd(
-                    keys=["image", "label"], channel_dim="no_channel"
+                    keys=["image", "label"], strict_check=False
                 ),
                 # RandSpatialCropSamplesd(
                 #     keys=["image", "label"],
@@ -280,7 +280,7 @@ def get_dataset(self, train_transforms):
         load_single_images = Compose(
             [
                 LoadImaged(keys=["image"]),
-                EnsureChannelFirstd(keys=["image"]),
+                EnsureChannelFirstd(keys=["image"], strict_check=False),
                 Orientationd(keys=["image"], axcodes="PLI"),
                 SpatialPadd(
                     keys=["image"],
@@ -1296,7 +1296,7 @@ def get_patch_loader_func(num_samples):
                 return Compose(
                     [
                         LoadImaged(keys=["image", "label"]),
-                        EnsureChannelFirstd(keys=["image", "label"]),
+                        EnsureChannelFirstd(keys=["image", "label"], strict_check=False),
                         RandSpatialCropSamplesd(
                             keys=["image", "label"],
                             roi_size=(
@@ -1381,7 +1381,7 @@ def get_patch_loader_func(num_samples):
                             # image_only=True,
                             # reader=WSIReader(backend="tifffile")
                         ),
-                        EnsureChannelFirstd(keys=["image", "label"]),
+                        EnsureChannelFirstd(keys=["image", "label"], strict_check=False),
                         Orientationd(keys=["image", "label"], axcodes="PLI"),
                         QuantileNormalizationd(keys=["image"]),
                         SpatialPadd(