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 (
str
,可选) – 数据集切分。支持"train"
(默认)和"val"
。size (
str
,可选) – 图像大小。支持"full"
(默认)、"320px"
和"160px"
。download (
bool
,可选) – 如果为True
,则下载数据集组件并将它们放置在root
中。已下载的归档文件不会再次下载。transform (可调用对象,可选) – 一个函数/转换,它接收 PIL 图像并返回转换后的版本,例如
transforms.RandomCrop
。target_transform – 一个函数/转换,它接收目标并对其进行转换。
- 特殊成员: