update transform timing

nnaakkaaii · Jul 4, 2024 · 76e0194 · 76e0194
1 parent e2c4d58
commit 76e0194
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/hrdae/conf b/hrdae/conf
diff --git a/hrdae/dataloaders/datasets/ct.py b/hrdae/dataloaders/datasets/ct.py
@@ -98,8 +98,6 @@ def __init__(
         if in_memory:
             for path in tqdm(self.paths, desc="loading datasets..."):
                 t = from_numpy(np.load(path)["arr_0"])
-                if transform is not None:
-                    t = transform(t)
                 self.data.append(t)
 
         self.slice_indexer = slice_indexer
@@ -119,8 +117,8 @@ def __getitem__(self, index: int) -> dict[str, Tensor]:
         else:  # not in memory
             assert not self.in_memory
             x_3d = from_numpy(np.load(str(self.paths[index]))["arr_0"])
-            if self.transform is not None:
-                x_3d = self.transform(x_3d)
+        if self.transform is not None:
+            x_3d = self.transform(x_3d)
 
         x_3d = x_3d.float()
         n, d, h, w = x_3d.size()