update ct

nnaakkaaii · Jul 12, 2024 · 472a4e6 · 472a4e6
1 parent a050aad
commit 472a4e6
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 3 deletions.
diff --git a/bin/mmnist/autoencoder2d/contrastive00/best_decoder.pth b/bin/mmnist/autoencoder2d/contrastive00/best_decoder.pth
diff --git a/bin/mmnist/autoencoder2d/contrastive00/best_encoder.pth b/bin/mmnist/autoencoder2d/contrastive00/best_encoder.pth
diff --git a/bin/mmnist/autoencoder2d/contrastive00/best_model.pth b/bin/mmnist/autoencoder2d/contrastive00/best_model.pth
diff --git a/hrdae/conf b/hrdae/conf
diff --git a/hrdae/dataloaders/datasets/ct.py b/hrdae/dataloaders/datasets/ct.py
@@ -98,6 +98,8 @@ 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
@@ -117,8 +119,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()