Skip to content
Snippets Groups Projects
Commit 5e28dd15 authored by ofhkr's avatar ofhkr
Browse files

added TODO in dataset class.

parent 9a9ada7d
No related branches found
No related tags found
1 merge request!47(Work in progress) Implementation of adaptive Dataset class which adapts to different data structures
......@@ -146,6 +146,7 @@ class Dataset(torch.utils.data.Dataset):
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 = []
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment