Disable strict checks for channelfirstd

Author: C-Achard

napari_cellseg3d/code_models/worker_training.py

@@ -201,7 +201,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"], channel_dim="no_channel", strict_check=False),
                 RandSpatialCropSamplesd(
                     keys=["image"],
                     roi_size=(
@@ -236,7 +236,7 @@ def get_dataset_eval(self, eval_dataset_dict):
             [
                 LoadImaged(keys=["image", "label"]),
                 EnsureChannelFirstd(
-                    keys=["image", "label"], channel_dim="no_channel"
+                    keys=["image", "label"], channel_dim="no_channel", strict_check=False
                 ),
                 # RandSpatialCropSamplesd(
                 #     keys=["image", "label"],