Python DatasetFromFolder Examples

Example #1

File: srcnn_data.py Project: mingshaofeng/pytorch_srcnn

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 #2

File: srcnn_data.py Project: hodgka/super_res_demo

def get_training_set(upscale_factor):

    root_dir = download_bsd300()
    # root_dir = '/media/alec/wd1/datasets/FLICKR30K'
    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 #3

def get_training_set(upscale_factor):

    root_dir = download_bsd300()
    #root_dir = "/home/mlvc04/imagenet"

    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))