proactive_IMD icon indicating copy to clipboard operation
proactive_IMD copied to clipboard

Published 20 hours ago • verified publisher icon vishal3477

Problems with dataset

Open stephenivy07 opened this issue 1 year ago • 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 !

stephenivy07 avatar Jan 17 '24 08:01 stephenivy07

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

vishal3477 avatar Feb 02 '24 12:02 vishal3477

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

stephenivy07 avatar Feb 03 '24 14:02 stephenivy07