def CreateDataset(opt): dataset = None if opt.dataset_mode == 'single': from data.single_dataset import SingleDataset dataset = SingleDataset() elif opt.dataset_mode == 'star' or opt.dataset_mode == 'n2': from data.half_dataset import N2Dataset dataset = N2Dataset() elif opt.dataset_mode == 'star1' or opt.dataset_mode == 'n2t': from data.half_dataset import N2TestDataset dataset = N2TestDataset() elif opt.dataset_mode == 'star2' or opt.dataset_mode == '2n': from data.half_dataset import _2NDataset dataset = _2NDataset() elif opt.dataset_mode == 'single_star' or opt.dataset_mode == 'sn2t': from data.single_dataset import SingleN2Dataset dataset = SingleN2Dataset() elif opt.dataset_mode == 'single_star1' or opt.dataset_mode == 'snt': from data.single_dataset import SingleNDataset dataset = SingleNDataset() else: raise ValueError("Dataset [%s] not recognized." % opt.dataset_mode) print("dataset [%s] was created" % (dataset.name())) dataset.initialize(opt) return dataset
def CreateSingleDataset(opt): dataset = None from data.single_dataset import SingleDataset dataset = SingleDataset() print("dataset [%s] was created" % (dataset.name())) dataset.initialize(opt) return dataset
def CreateDataset(opt): dataset = None if opt.dataset_mode == 'single': from data.single_dataset import SingleDataset dataset = SingleDataset() elif opt.dataset_mode == 'aligned': from data.aligned_dataset import AlignedDataset dataset = AlignedDataset() else: raise ValueError("Dataset [%s] not recognized." % opt.dataset_mode) print(("dataset [%s] was created" % (dataset.name()))) dataset.initialize(opt) return dataset