Python UnalignedDataset.name Examples

Example #1

File: custom_dataset_data_loader.py Project: xiaoyongy/PhotoSketchMAN

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

Example #2

File: custom_dataset_data_loader.py Project: Shade5/Recycle-GAN

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

Example #3

File: custom_dataset_data_loader.py Project: KarelZhang/CSDNet-CSDGAN

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

Example #4

File: custom_dataset_data_loader.py Project: changruowang/MyEnhanceProj

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

Example #5

def CreateDataset(opt):
    dataset = UnalignedDataset()

    print("dataset [%s] was created" % (dataset.name()))
    dataset.initialize(opt)
    return dataset