diff --git a/README.md b/README.md index 10954b7..dbbfd25 100644 --- a/README.md +++ b/README.md @@ -75,6 +75,36 @@ pip install tensorboardX pip install terminaltables ``` +## Dataset Preparation + +Download the [ImageNet-1K](http://image-net.org/) classification dataset and structure the data as follows: +``` +/path/to/imagenet-1k/ + train/ + class1/ + img1.jpeg + class2/ + img2.jpeg + val/ + class1/ + img3.jpeg + class2/ + img4.jpeg +``` + +For pre-training on [ImageNet-22K](http://image-net.org/), download the dataset and structure the data as follows: +``` +/path/to/imagenet-22k/ + class1/ + img1.jpeg + class2/ + img2.jpeg + class3/ + img3.jpeg + class4/ + img4.jpeg +``` + ## Testing We give an example evaluation command for VanillaNet-5: