FileNotFoundError

[zenglingqi647]

Hi!

I tried downloading the ImageNet-Video-Robust dataset from the repository. However, the data are organized in this way:

And when running the code, the following error would raise:

Test on the original test set
Preparing data...
Traceback (most recent call last):
File "main.py", line 37, in
_, trloader = prepare_train_data(args)
File "/scratch/PI/asd/asdfg/TTA/ttt_imagenet_release/utils/train_helpers.py", line 90, in prepare_train_data
rotation_transform=rotation_tr_transforms)
File "/scratch/PI/asd/asdfg/TTA/ttt_imagenet_release/utils/rotation.py", line 42, in init
super(RotateImageFolder, self).init(traindir, train_transform)
File "/scratch/PI/asd/asdfg/miniconda3/envs/ttt-icml20/lib/python3.7/site-packages/torchvision/datasets/folder.py", line 315, in init
is_valid_file=is_valid_file,
File "/scratch/PI/asd/asdfg/miniconda3/envs/ttt-icml20/lib/python3.7/site-packages/torchvision/datasets/folder.py", line 144, in init
classes, class_to_idx = self.find_classes(self.root)
File "/scratch/PI/asd/asdfg/miniconda3/envs/ttt-icml20/lib/python3.7/site-packages/torchvision/datasets/folder.py", line 218, in find_classes
return find_classes(directory)
File "/scratch/PI/asd/asdfg/miniconda3/envs/ttt-icml20/lib/python3.7/site-packages/torchvision/datasets/folder.py", line 40, in find_classes
classes = sorted(entry.name for entry in os.scandir(directory) if entry.is_dir())
FileNotFoundError: [Errno 2] No such file or directory: '/home/asdfg/scratch/asd/asdfg/TTA/data/imagenet/train'

The data directory, /home/asdfg/scratch/asd/asdfg/TTA/data/imagenet, has structure like this:

How should the data be organized, or where should the 'train' come from?

Thank you!