vishal3477/proactive_IMD

Problems with dataset

Opened this issue · 2 comments

Hi, I met some problems when I directly specify the path of --data_train or --data_test.
My codes is:
parser.add_argument('--data_train',default='./data/CELEBA_HQ',help='root directory for training data')

And I got errors:
RuntimeError: Found 0 files in subfolders of: ./data/CELEBA_HQ
Supported extensions are: .jpg,.jpeg,.png,.ppm,.bmp,.pgm,.tif,.tiff,.webp

Could you please tell me in detail how to organize the data? Thank you !

Hi, you need to have a structure like this: ./CELEBA_HQ/images/***

Thank you! I tried to organize the data, but met new problems:
I download all the pre-trained checkpoints and run test_stargan.py, when I set the --set_size as 4(default),I got this error:
Snipaste_2024-02-03_22-34-58

When I set the --set_size as 3, I got this error:
Snipaste_2024-02-03_22-34-16