ImageNet — Torchvision main documentation - pytorch.org
pytorch.org › torchvisionclass torchvision.datasets.ImageNet(root: str, split: str = 'train', download: Optional[str] = None, **kwargs: Any) [source] ImageNet 2012 Classification Dataset. Parameters. root ( string) – Root directory of the ImageNet Dataset. split ( string, optional) – The dataset split, supports train, or val. transform ( callable, optional) – A ...