From b8341337d90c93011e6cb1f17316cd1bd4ca6db6 Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Tue, 4 Oct 2022 11:45:40 +0200
Subject: [PATCH] patch dataset now correctly uses augemntation transforms
---
mu_map/dataset/patches.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/mu_map/dataset/patches.py b/mu_map/dataset/patches.py
index 530e61a..9aade94 100644
--- a/mu_map/dataset/patches.py
+++ b/mu_map/dataset/patches.py
@@ -65,6 +65,8 @@ class MuMapPatchDataset(MuMapDataset):
recon = recon[:, 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
def __len__(self):
--
GitLab