load_dataset#
- deepinv.utils.load_dataset(dataset_name, transform, data_dir=None, download=True, url=None, file_type='zip')[source]#
Loads an ImageFolder dataset from DeepInverse HuggingFace repository.
- Parameters:
dataset_name (str, pathlib.Path) – dataset name without file extension.
transform (Callable) – optional transform to pass to torchvision dataset.
data_dir (str, pathlib.Path) – dataset root directory, defaults to None
download (bool) – whether to download, defaults to True
url (str) – download URL, if
None
, gets URL usingdeepinv.utils.get_image_url()
file_type (str) – file extension, defaults to “zip”
- Returns:
torchvision ImageFolder dataset.
- Return type:
Examples using load_dataset
:#

Regularization by Denoising (RED) for Super-Resolution.
Regularization by Denoising (RED) for Super-Resolution.

Deep Equilibrium (DEQ) algorithms for image deblurring
Deep Equilibrium (DEQ) algorithms for image deblurring

Unfolded Chambolle-Pock for constrained image inpainting
Unfolded Chambolle-Pock for constrained image inpainting