Skip to content

Commit fc377d0

Browse files
puhukNicolasHug
andauthored
Improve ImageNet documentation (#7503)
Co-authored-by: Nicolas Hug <[email protected]> Co-authored-by: Nicolas Hug <[email protected]>
1 parent 2975710 commit fc377d0

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

torchvision/datasets/imagenet.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,12 @@
2121
class ImageNet(ImageFolder):
2222
"""`ImageNet <http://image-net.org/>`_ 2012 Classification Dataset.
2323
24+
.. note::
25+
Before using this class, it is required to download ImageNet 2012 dataset from
26+
`here <https://image-net.org/challenges/LSVRC/2012/2012-downloads.php>`_ and
27+
place the files ``ILSVRC2012_devkit_t12.tar.gz`` and ``ILSVRC2012_img_train.tar``
28+
or ``ILSVRC2012_img_val.tar`` based on ``split`` in the root directory.
29+
2430
Args:
2531
root (string): Root directory of the ImageNet Dataset.
2632
split (string, optional): The dataset split, supports ``train``, or ``val``.

0 commit comments

Comments
 (0)