Caltech101 — Torchvision 0.24 documentation (original) (raw)

class torchvision.datasets.Caltech101(root: Union[str, Path], 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 (str or pathlib.Path) – Root directory of dataset where directorycaltech101 exists or will be saved to if download is set to True.
• target_type (string or list, optional) – Type of target to use, category orannotation. Can also be a list to output a tuple with all specified target types. category represents the target class, andannotation 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