BR-IDL / PaddleViT

:robot: PaddleViT: State-of-the-art Visual Transformer and MLP Models for PaddlePaddle 2.0+
https://github.com/BR-IDL/PaddleViT
Apache License 2.0
1.22k stars 318 forks source link

Add automatic download for imagenet list files. #171

Open xperzy opened 2 years ago

xperzy commented 2 years ago

Describe your feature request Now the dataset for ImageNet2012 requires the "train_list.txt" and "val_list.txt" in the same folder as "train" and "val" which stores the images.

Users are required to download the txt files and locate to the correct dir manually, which can be automated by our code.

Describe the possible solution

  1. txt files can be uploaded in github repo
  2. Add check and download code in Imagenet Dataset class