(upscale_factor)
| 51 | |
| 52 | |
| 53 | def get_training_set(upscale_factor): |
| 54 | root_dir = download_bsd300() |
| 55 | train_dir = join(root_dir, "train") |
| 56 | crop_size = calculate_valid_crop_size(256, upscale_factor) |
| 57 | |
| 58 | return DatasetFromFolder(train_dir, |
| 59 | input_transform=input_transform(crop_size, upscale_factor), |
| 60 | target_transform=target_transform(crop_size)) |
| 61 | |
| 62 | |
| 63 | def get_test_set(upscale_factor): |
no test coverage detected