NVlabs / RED-diff

Other
60 stars 8 forks source link

Question for dataset #1

Open DavidYan2001 opened 10 months ago

DavidYan2001 commented 10 months ago

Dear author,

When I want to reproduce your project code, in the evaluation as mentioned in your README.md file, it displays that " The meta file /home/mmardani/research/datasets/imagenet-root/imagenet/meta.bin is not present in the root directory or is corrupted. This file is automatically created by the ImageNet dataset."

I find that in files like imagenet256_val.yaml, the variables "root, meta_root, subset_txt" are predefined by your own project, may I ask what imagenet-root contains and what meta.bin stands for?

man-sean commented 9 months ago

As far as I can tell this is the standard ImageNet dataset as described by torchvision. meta.bin is created automatically by the Imagenet class if ILSVRC2012_devkit_t12.tar.gz is presented in the root directory. This file can be downloaded from here: https://image-net.org/challenges/LSVRC/2012/2012-downloads.php

Note that if your imagenet directory is at the path \some\path\imagenet than the root variable need to be:

  1. \some\path if split is train or val
  2. \some\path\imagenet otherwise

All the logic that I described here is presented in the file https://github.com/NVlabs/RED-diff/blob/master/datasets/imagenet.py