We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d2a9399 commit 4babcffCopy full SHA for 4babcff
1 file changed
napari_cellseg3d/model_workers.py
@@ -398,19 +398,32 @@ def load_layer(self):
398
399
# print(volume.shape)
400
# print(volume.dtype)
401
-
402
- load_transforms = Compose(
+ if self.use_window:
+ load_transforms = Compose(
403
[
404
ToTensor(),
405
# anisotropic_transform,
406
AddChannel(),
407
- SpatialPad(spatial_size=pad),
+ # SpatialPad(spatial_size=pad),
408
409
EnsureType(),
410
],
411
map_items=False,
412
log_stats=True,
413
)
414
+ else:
415
416
+ [
417
+ ToTensor(),
418
+ # anisotropic_transform,
419
+ AddChannel(),
420
+ SpatialPad(spatial_size=pad),
421
422
+ EnsureType(),
423
+ ],
424
+ map_items=False,
425
+ log_stats=True,
426
+ )
427
428
self.log("\nLoading dataset...")
429
input_image = load_transforms(volume)
0 commit comments