ImageNet¶
- class torchvision.datasets.ImageNet(root: Union[str, Path], split: str = 'train', **kwargs: Any)[source]¶
ImageNet 2012 分类数据集。
注意
在使用此类之前,需要从这里下载 ImageNet 2012 数据集,并将文件
ILSVRC2012_devkit_t12.tar.gz
和ILSVRC2012_img_train.tar
或ILSVRC2012_img_val.tar
(根据split
)放置在根目录中。- 参数:
root (str 或
pathlib.Path
) – ImageNet 数据集的根目录。split (字符串, 可选) – 数据集分割,支持
train
或val
。transform (可调用对象, 可选) – 一个函数/转换,它接收一个 PIL 图像并返回一个转换后的版本。例如,
transforms.RandomCrop
target_transform (可调用对象, 可选) – 一个函数/转换,它接收目标并对其进行转换。
loader – 一个函数,用于根据路径加载图像。
- 特殊成员: