def CreateDataset(opt): dataset = None from data.unaligned_dataset import UnalignedDataset dataset = UnalignedDataset() print("dataset [%s] was created" % (dataset.name())) dataset.initialize(opt) return dataset
def CreateDataset(opt): if opt.dataset_mode == 'unaligned': from data.unaligned_dataset import UnalignedDataset dataset = UnalignedDataset() elif opt.dataset_mode == 'unaligned_triplet': from data.unaligned_triplet_dataset import UnalignedTripletDataset dataset = UnalignedTripletDataset() else: raise ValueError("Dataset [%s] not recognized." % opt.dataset_mode) print("dataset [%s] was created" % (dataset.name())) dataset.initialize(opt) return dataset
def CreateDataset(opt): dataset = None if opt.dataset_mode == 'unpair': from data.unaligned_dataset import UnalignedDataset dataset = UnalignedDataset() elif opt.dataset_mode == 'pair': from data.pair_dataset import PairDataset dataset = PairDataset() elif opt.dataset_mode == 'test': from data.test_dataset import TestDataset dataset = TestDataset() else: raise ValueError("Dataset [%s] not recognized." % opt.dataset_mode) print("dataset [%s] was created" % (dataset.name())) dataset.initialize(opt) return dataset
def CreateDataset(opt): ''' 创建数据集 目前支持两种数据集 Unaligned: 用于直接读取 图片文件 在测试的时候使用 hd5f: 用于训练的时候读取 预先制作的 .h5 格式的训练数据集 ''' dataset = None if opt.dataset_mode == 'unaligned': from data.unaligned_dataset import UnalignedDataset dataset = UnalignedDataset() elif opt.dataset_mode == 'hd5f': from data.hd5f_dataset import HD5FDataset dataset = HD5FDataset() else: raise ValueError("Dataset [%s] not recognized." % opt.dataset_mode) print("dataset [%s] was created" % (dataset.name())) dataset.initialize(opt) return dataset
def CreateDataset(opt): dataset = UnalignedDataset() print("dataset [%s] was created" % (dataset.name())) dataset.initialize(opt) return dataset