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 using deepinv.utils.get_image_url()
file_type (str) – file extension, defaults to “zip”

Returns: