Is there any requirement on the training images?

[wd255]

Hi, thanks for the great job!

When I try it out, I used a subfolder of imagenet (/ILSVRC/Data/CLS-LOC/train/n02096437) which contains a lot of images as dataset_folder, but I got the error: 0 training samples found at /ILSVRC/Data/CLS-LOC/train/n02096437

I double checked the folder, it has a lot of images.

I wonder if there's any requirement?

[lucidrains]

hello! hmm that's weird, could you check what extensions the images are?

[wd255]

Hi @lucidrains thanks for the timely response. I checked the code and found it's because the extension of the image files in my imagnet dataset was "JPEG" instead of "jpeg". This is the official magnet dataset so I assume others have the same issue, so I added #20

[lucidrains]

@wd255 put in a quick hack, let me know that works!