Imagenette¶
- class torchvision.datasets.Imagenette(root: Union[str, Path], split: str = 'train', size: str = 'full', download=False, transform: Optional[Callable] = None, target_transform: Optional[Callable] = None)[source]¶
Imagenette 图像分类数据集。
- 参数:
root (str 或
pathlib.Path
) – Imagenette 数据集的根目录。split (string, optional) – 数据集拆分。支持
"train"
(默认) 和"val"
。size (string, optional) – 图像大小。支持
"full"
(默认),"320px"
, 和"160px"
。download (bool, optional) – 如果为
True
, 则下载数据集组件并将它们放置在root
中。已下载的存档不会再次下载。transform (callable, optional) – 接收 PIL 图像并返回变换后版本(例如
transforms.RandomCrop
)的函数/变换。target_transform – 接收目标并对其进行变换的函数/变换。
- 特殊成员: