Normalize before ToTensor to avoid OOM from torch

Author: C-Achard

napari_cellseg3d/code_models/worker_inference.py

@@ -244,11 +244,11 @@ def load_layer(self):
         if self.config.sliding_window_config.is_enabled():
             load_transforms = Compose(
                 [
+                    QuantileNormalization(),
                     ToTensor(),
                     # anisotropic_transform,
                     AddChannel(),
                     # SpatialPad(spatial_size=pad),
-                    QuantileNormalization(),
                     AddChannel(),
                     EnsureType(),
                 ],
@@ -260,10 +260,10 @@ def load_layer(self):
             pad = utils.get_padding_dim(dims_check)
             load_transforms = Compose(
                 [
+                    QuantileNormalization(),
                     ToTensor(),
                     # anisotropic_transform,
                     AddChannel(),
-                    QuantileNormalization(),
                     SpatialPad(spatial_size=pad),
                     AddChannel(),
                     EnsureType(),