Caltech101¶
- 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)[源代码]¶
Caltech 101 数据集。
警告
此类需要 scipy 从 .mat 格式加载目标文件。
- 参数::
root (str 或
pathlib.Path
) – 数据集的根目录,其中包含目录caltech101
,如果将 download 设置为 True,则将其保存到该目录中。target_type (字符串 或 列表, 可选) – 要使用的目标类型,
category
或annotation
。也可以是列表,以输出包含所有指定目标类型的元组。category
表示目标类别,而annotation
是从手工生成的轮廓中提取的一组点。默认值为category
。transform (可调用对象, 可选) – 一个函数/转换,它接收 PIL 图像并返回一个转换后的版本。例如,
transforms.RandomCrop
target_transform (可调用对象, 可选) – 一个函数/转换,它接收目标并对其进行转换。
download (布尔值, 可选) –
如果为 True,则从互联网下载数据集并将其放入根目录中。如果数据集已经下载,则不会再次下载。
警告
要下载数据集,需要 gdown。
- 特殊成员::