Commit 4ec53493 authored by Alessia Marcolini's avatar Alessia Marcolini
Browse files

Return also patient

parent 479b865b
......@@ -39,7 +39,13 @@ class TileDataset(BaseDataset):
image_out = self.__transform__(image)
label = self.labels[i]
sample = {'image': image_out, 'label': label, 'filename': filename}
patient = self.patients[i]
sample = {
'image': image_out,
'label': label,
'filename': filename,
'patient': patient,
return sample
def __shuffle__(self):
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment