Exemplo n.º 1
0
 def load_writeable_dataset(self, ds, path):
     if self.isMultiOutput():
         rr = utils.load(path)
         resName = ds.get_name() + "_predictions"
         result = datasets.BufferedWriteableDS(ds, resName, path, rr)
     else:
         rr = np.load(path)
         resName = ds.get_name() + "_predictions"
         result = datasets.BufferedWriteableDS(ds, resName, path, rr)
     return result
Exemplo n.º 2
0
 def create_writeable_dataset(self, dataset: datasets.DataSet,
                              dsPath: str) -> datasets.WriteableDataSet:
     inp, output = utils.load_yaml(self.path + ".shapes")
     resName = dataset.get_name() + "_predictions"
     result = datasets.BufferedWriteableDS(dataset,
                                           resName,
                                           dsPath,
                                           pickle=self.isMultiOutput())
     return result