Ejemplo n.º 1
0
 def __init__(self, dataframe, sequences, transform=None):
     Dataset.__init__(self)
     NiftiDataset.__init__(self)
     self.df = dataframe
     self.sequences = sequences
     self.patients = self.df.index.values
     self.transform = transform
Ejemplo n.º 2
0
 def __init__(self,
              df=None,
              brainmask=True,
              segmentation=True,
              sequences=('t1', 'flair', 't2', 't1Gd')):
     NiftiDataset.__init__(self)
     if df is not None:
         self.df = df
     self.brainmask = brainmask
     self.segmentation = segmentation
     self.sequences = sequences
 def __init__(self, dataframe, sequences, transform=None, preprocess_config=None):
     Dataset.__init__(self)
     NiftiDataset.__init__(self)
     self.df = dataframe
     self.sequences = sequences
     self.patients = self.df.index.values
     print(self.df['seg'][0])
     print(self.df['seg'][1])
     print(self.df['seg'][2])
     self.transform = transform
     self.config_file = preprocess_config
Ejemplo n.º 4
0
 def __init__(self,
              dataframe,
              sequences,
              transform=None,
              brainmask=True,
              segmentation=True):
     Dataset.__init__(self)
     NiftiDataset.__init__(self)
     self.brainmask = brainmask
     self.df = dataframe
     self.sequences = sequences
     self.patients = self.df.index.values
     self.transform = transform
     self.segmentation = segmentation
Ejemplo n.º 5
0
 def __init__(self, image_paths, brainmask_path=None, transform=None):
     NiftiDataset.__init__(self)
     self.image_paths = image_paths
     self.brainmask_path = brainmask_path
     self.transform = transform
Ejemplo n.º 6
0
 def __init__(self, df=None):
     NiftiDataset.__init__(self)
     if df is not None:
         self.df = df
Ejemplo n.º 7
0
 def __init__(self, df=None, sequences=('t1', 'flair', 't2', 't1Gd')):
     NiftiDataset.__init__(self)
     if df is not None:
         self.df = df
     self.sequences = sequences