diff --git a/torchvision/datasets/imagenet.py b/torchvision/datasets/imagenet.py index 4b86bf2f2b9..eee0d005488 100644 --- a/torchvision/datasets/imagenet.py +++ b/torchvision/datasets/imagenet.py @@ -21,6 +21,12 @@ class ImageNet(ImageFolder): """`ImageNet `_ 2012 Classification Dataset. + .. note:: + Before using this class, it is required to download ImageNet 2012 dataset from + `here `_ and + place the files ``ILSVRC2012_devkit_t12.tar.gz`` and ``ILSVRC2012_img_train.tar`` + or ``ILSVRC2012_img_val.tar`` based on ``split`` in the root directory. + Args: root (string): Root directory of the ImageNet Dataset. split (string, optional): The dataset split, supports ``train``, or ``val``.