From f722137fe8fddf44df0b82937df928c14aa6456a Mon Sep 17 00:00:00 2001 From: C-Achard Date: Sat, 21 Dec 2024 12:02:30 +0100 Subject: [PATCH] Update worker_training.py --- napari_cellseg3d/code_models/worker_training.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/napari_cellseg3d/code_models/worker_training.py b/napari_cellseg3d/code_models/worker_training.py index 1fb15f2b..d393868d 100644 --- a/napari_cellseg3d/code_models/worker_training.py +++ b/napari_cellseg3d/code_models/worker_training.py @@ -281,7 +281,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"],