Skip to content

Commit f722137

Browse files
committed
Update worker_training.py
1 parent a76037c commit f722137

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

napari_cellseg3d/code_models/worker_training.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -281,7 +281,11 @@ def get_dataset(self, train_transforms):
281281
load_single_images = Compose(
282282
[
283283
LoadImaged(keys=["image"]),
284-
EnsureChannelFirstd(keys=["image"], channel_dim="no_channel", strict_check=False),
284+
EnsureChannelFirstd(
285+
keys=["image"],
286+
channel_dim="no_channel",
287+
strict_check=False,
288+
),
285289
Orientationd(keys=["image"], axcodes="PLI"),
286290
SpatialPadd(
287291
keys=["image"],

0 commit comments

Comments
 (0)