Shortcuts

Caltech101

class torchvision.datasets.Caltech101(root: str, target_type: Union[List[str], str] = 'category', transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, download: bool = False)[source]

Caltech 101 Dataset.

Warning

This class needs scipy to load target files from .mat format.

Parameters:
  • root (string) – Root directory of dataset where directory caltech101 exists or will be saved to if download is set to True.

  • target_type (string or list, optional) – Type of target to use, category or annotation. Can also be a list to output a tuple with all specified target types. category represents the target class, and annotation is a list of points from a hand-generated outline. Defaults to category.

  • transform (callable, optional) – A function/transform that takes in a PIL image and returns a transformed version. E.g, transforms.RandomCrop

  • target_transform (callable, optional) – A function/transform that takes in the target and transforms it.

  • download (bool, optional) –

    If true, downloads the dataset from the internet and puts it in root directory. If dataset is already downloaded, it is not downloaded again.

    Warning

    To download the dataset gdown is required.

Special-members:

__getitem__(index: int) Tuple[Any, Any][source]
Parameters:

index (int) – Index

Returns:

(image, target) where the type of target specified by target_type.

Return type:

tuple

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources