Python data.custom_dataset_data_loader.CustomDatasetDataLoader() Examples

The following are 14 code examples of data.custom_dataset_data_loader.CustomDatasetDataLoader(). You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may also want to check out all available functions/classes of the module data.custom_dataset_data_loader , or try the search function .
Example #1
Source File: train.py    From GANimation with GNU General Public License v3.0 6 votes vote down vote up 
def __init__(self):
        self._opt = TrainOptions().parse()
        data_loader_train = CustomDatasetDataLoader(self._opt, is_for_train=True)
        data_loader_test = CustomDatasetDataLoader(self._opt, is_for_train=False)

        self._dataset_train = data_loader_train.load_data()
        self._dataset_test = data_loader_test.load_data()

        self._dataset_train_size = len(data_loader_train)
        self._dataset_test_size = len(data_loader_test)
        print('#train images = %d' % self._dataset_train_size)
        print('#test images = %d' % self._dataset_test_size)

        self._model = ModelsFactory.get_by_name(self._opt.model, self._opt)
        self._tb_visualizer = TBVisualizer(self._opt)

        self._train()
Example #2
Source File: data_loader.py    From Recycle-GAN with MIT License 5 votes vote down vote up 
def CreateDataLoader(opt):
    from data.custom_dataset_data_loader import CustomDatasetDataLoader
    data_loader = CustomDatasetDataLoader()
    print(data_loader.name())
    data_loader.initialize(opt)
    return data_loader
Example #3
Source File: data_loader.py    From DepthAwareCNN with MIT License 5 votes vote down vote up 
def CreateDataLoader(opt):
    from data.custom_dataset_data_loader import CustomDatasetDataLoader
    data_loader = CustomDatasetDataLoader()
    print(data_loader.name())
    data_loader.initialize(opt)
    return data_loader
Example #4
Source File: data_loader.py    From iSketchNFill with GNU General Public License v3.0 5 votes vote down vote up 
def CreateDataLoader(opt):
    from data.custom_dataset_data_loader import CustomDatasetDataLoader
    data_loader = CustomDatasetDataLoader()
    print(data_loader.name())
    data_loader.initialize(opt)
    return data_loader
Example #5
Source File: data_loader.py    From everybody_dance_now_pytorch with GNU Affero General Public License v3.0 5 votes vote down vote up 
def CreateDataLoader(opt):
    from data.custom_dataset_data_loader import CustomDatasetDataLoader
    data_loader = CustomDatasetDataLoader()
    print(data_loader.name())
    data_loader.initialize(opt)
    return data_loader
Example #6
Source File: data_loader.py    From non-stationary_texture_syn with MIT License 5 votes vote down vote up 
def CreateDataLoader(opt):
    from data.custom_dataset_data_loader import CustomDatasetDataLoader
    data_loader = CustomDatasetDataLoader()
    print(data_loader.name())
    data_loader.initialize(opt)
    return data_loader
Example #7
Source File: data_loader.py    From EverybodyDanceNow_reproduce_pytorch with MIT License 5 votes vote down vote up 
def CreateDataLoader(opt):
    from data.custom_dataset_data_loader import CustomDatasetDataLoader
    data_loader = CustomDatasetDataLoader()
    print(data_loader.name())
    data_loader.initialize(opt)
    return data_loader
Example #8
Source File: data_loader.py    From 2.5D-Visual-Sound with Creative Commons Attribution 4.0 International 5 votes vote down vote up 
def CreateDataLoader(opt):
    from data.custom_dataset_data_loader import CustomDatasetDataLoader
    data_loader = CustomDatasetDataLoader()
    print(data_loader.name())
    data_loader.initialize(opt)
    return data_loader
Example #9
Source File: data_loader.py    From EverybodyDanceNow-Temporal-FaceGAN with MIT License 5 votes vote down vote up 
def CreateDataLoader(opt):
    from data.custom_dataset_data_loader import CustomDatasetDataLoader
    data_loader = CustomDatasetDataLoader()
    print(data_loader.name())
    data_loader.initialize(opt)
    return data_loader
Example #10
Source File: data_loader.py    From deep-learning-for-document-dewarping with MIT License 5 votes vote down vote up 
def CreateDataLoader(opt):
    from data.custom_dataset_data_loader import CustomDatasetDataLoader
    data_loader = CustomDatasetDataLoader()
    print(data_loader.name())
    data_loader.initialize(opt)
    return data_loader
Example #11
Source File: data_loader.py    From PerceptualSimilarity with BSD 2-Clause "Simplified" License 5 votes vote down vote up 
def CreateDataLoader(datafolder,dataroot='./dataset',dataset_mode='2afc',load_size=64,batch_size=1,serial_batches=True,nThreads=4):
    from data.custom_dataset_data_loader import CustomDatasetDataLoader
    data_loader = CustomDatasetDataLoader()
    # print(data_loader.name())
    data_loader.initialize(datafolder,dataroot=dataroot+'/'+dataset_mode,dataset_mode=dataset_mode,load_size=load_size,batch_size=batch_size,serial_batches=serial_batches, nThreads=nThreads)
    return data_loader
Example #12
Source File: data_loader.py    From Shift-Net_pytorch with MIT License 5 votes vote down vote up 
def CreateDataLoader(opt):
    from data.custom_dataset_data_loader import CustomDatasetDataLoader
    data_loader = CustomDatasetDataLoader()
    print(data_loader.name())
    data_loader.initialize(opt)
    return data_loader
Example #13
Source File: data_loader.py    From Bayesian-CycleGAN with MIT License 5 votes vote down vote up 
def CreateDataLoader(opt):
    from data.custom_dataset_data_loader import CustomDatasetDataLoader
    data_loader = CustomDatasetDataLoader()
    print(data_loader.name())
    data_loader.initialize(opt)
    return data_loader
Example #14
Source File: data_loader.py    From neurips18_hierchical_image_manipulation with MIT License 5 votes vote down vote up 
def CreateDataLoader(opt):
    data_loader = CustomDatasetDataLoader()
    print(data_loader.name())
    data_loader.initialize(opt)
    return data_loader