Python dataset.DatasetFromFolder() Examples

The following are 9 code examples of dataset.DatasetFromFolder(). 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 dataset , or try the search function .
Example #1
Source File: data.py    From iSeeBetter with MIT License 5 votes vote down vote up 
def get_training_set(data_dir, nFrames, upscale_factor, data_augmentation, file_list, other_dataset, patch_size, future_frame):
    print("Training samples chosen:", file_list)
    return DatasetFromFolder(data_dir,nFrames, upscale_factor, data_augmentation, file_list, other_dataset, patch_size,future_frame,
                             transform=transform())
Example #2
Source File: data.py    From iSeeBetter with MIT License 5 votes vote down vote up 
def get_eval_set(data_dir, nFrames, upscale_factor, data_augmentation, file_list, other_dataset, patch_size, future_frame):
    return DatasetFromFolder(data_dir,nFrames, upscale_factor, data_augmentation, file_list, other_dataset, patch_size,future_frame,
                             transform=transform())
Example #3
Source File: data.py    From aerial_mtl with BSD 3-Clause "New" or "Revised" License 5 votes vote down vote up 
def get_training_set(upscale_factor):
    root_dir = download_bsd300()
    train_dir = join(root_dir, "train")
    crop_size = calculate_valid_crop_size(256, upscale_factor)

    return DatasetFromFolder(root_dir, 'train',
                             input_transform=input_transform(crop_size, upscale_factor),
                             target_transform=target_transform(crop_size))
Example #4
Source File: data.py    From aerial_mtl with BSD 3-Clause "New" or "Revised" License 5 votes vote down vote up 
def get_test_set(upscale_factor):
    root_dir = download_bsd300()
    test_dir = join(root_dir, "test")
    crop_size = calculate_valid_crop_size(256, upscale_factor)

    return DatasetFromFolder(root_dir, 'test',
                             input_transform=input_transform(crop_size, upscale_factor),
                             target_transform=target_transform(crop_size))
Example #5
Source File: data.py    From examples with BSD 3-Clause "New" or "Revised" License 5 votes vote down vote up 
def get_training_set(upscale_factor):
    root_dir = download_bsd300()
    train_dir = join(root_dir, "train")
    crop_size = calculate_valid_crop_size(256, upscale_factor)

    return DatasetFromFolder(train_dir,
                             input_transform=input_transform(crop_size, upscale_factor),
                             target_transform=target_transform(crop_size))
Example #6
Source File: data.py    From examples with BSD 3-Clause "New" or "Revised" License 5 votes vote down vote up 
def get_test_set(upscale_factor):
    root_dir = download_bsd300()
    test_dir = join(root_dir, "test")
    crop_size = calculate_valid_crop_size(256, upscale_factor)

    return DatasetFromFolder(test_dir,
                             input_transform=input_transform(crop_size, upscale_factor),
                             target_transform=target_transform(crop_size))
Example #7
Source File: data.py    From PyTorch with MIT License 5 votes vote down vote up 
def get_training_set(upscale_factor):
    root_dir = download_bsd300()
    train_dir = join(root_dir, "train")
    crop_size = calculate_valid_crop_size(256, upscale_factor)

    return DatasetFromFolder(train_dir,
                             input_transform=input_transform(crop_size, upscale_factor),
                             target_transform=target_transform(crop_size))
Example #8
Source File: data.py    From PyTorch with MIT License 5 votes vote down vote up 
def get_test_set(upscale_factor):
    root_dir = download_bsd300()
    test_dir = join(root_dir, "test")
    crop_size = calculate_valid_crop_size(256, upscale_factor)

    return DatasetFromFolder(test_dir,
                             input_transform=input_transform(crop_size, upscale_factor),
                             target_transform=target_transform(crop_size))
Example #9
Source File: data.py    From DBPN-Pytorch with MIT License 5 votes vote down vote up 
def get_training_set(data_dir, hr, upscale_factor, patch_size, data_augmentation):
    hr_dir = join(data_dir, hr)
    return DatasetFromFolder(hr_dir,patch_size, upscale_factor, data_augmentation,
                             transform=transform())