Pytorch data loader

DataLoader

class MRIDataset(Dataset):
    
    def __init__(self, root_dir, labels, transform=None):

        self.root_dir = root_dir        
        self.transform = transform     
        self.directories = []
        self.len = 0
        self.labels = labels
        
        self.clin_data = pd.read_csv("../data4/lon_clin.csv")
              
        train_dirs = []
        for label in labels:
            train_dirs.append(root_dir + label)
        
        for dir in train_dirs:
            for path in glob.glob(dir + "/*"):
                self.directories.append(pathlib.Path(path))
                
        self.len = len(self.directories)
        
        
    def __len__(self):
        
        return self.len