From 5e28dd15fd794a6a34068cf78014a2253aa831ce Mon Sep 17 00:00:00 2001
From: s184364 <s184364@student.dtu.dk>
Date: Thu, 25 Jan 2024 09:32:10 +0100
Subject: [PATCH] added TODO in dataset class.
---
qim3d/utils/data.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/qim3d/utils/data.py b/qim3d/utils/data.py
index e50a3b87..50f61554 100644
--- a/qim3d/utils/data.py
+++ b/qim3d/utils/data.py
@@ -145,7 +145,8 @@ class Dataset(torch.utils.data.Dataset):
NotImplementedError: in Case 3, if a folder is found among the list of files.
NotImplementedError: If the data structure does not fall into one of the three cases.
'''
-
+
+ # TODO: make sure user can specify extra mask names (add in prepare_datasets()?)
target_folder_names = ['mask','label','target']
# Case 1
@@ -186,6 +187,7 @@ class Dataset(torch.utils.data.Dataset):
self.sample_targets = [target for target in sorted(Path(self.root_path,targets).iterdir())]
# Case 3
+ # TODO: Make sure the split keeps the data from the same patient in either train, val or predict.
elif self.n_folders > 2:
sample_images = []
sample_targets = []
--
GitLab