lucidrains/magvit2-pytorch

Is there any requirement on the training images?

wd255 opened this issue · 3 comments

wd255 commented

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 commented

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

wd255 commented

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 commented

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