Skip to content
Snippets Groups Projects
Commit b8341337 authored by Tamino Huxohl's avatar Tamino Huxohl
Browse files

patch dataset now correctly uses augemntation transforms

parent 455bed65
Branches
Tags
No related merge requests found
Hide whitespace changes
Inline Side-by-side
mu_map/dataset/patches.py
+ 2
− 0
View file @ b8341337
...@@ -65,6 +65,8 @@ class MuMapPatchDataset(MuMapDataset): ...@@ -65,6 +65,8 @@ class MuMapPatchDataset(MuMapDataset):
recon = recon[:, z : z + s, y : y + s, x : x + s] recon = recon[:, z : z + s, y : y + s, x : x + s]
mu_map = mu_map[:, z : z + s, y : y + s, x : x + s] mu_map = mu_map[:, z : z + s, y : y + s, x : x + s]
recon, mu_map = self.transform_augmentation(recon, mu_map)
return recon, mu_map return recon, mu_map
def __len__(self): def __len__(self):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment