Some problem with tqdm(dataloader). Somehow it couldn't load the image properly about self-correction-human-parsing HOT 3 CLOSED

After some trouble shootings, the problem was caused by Jupyter checkpoint in the input directory. This issue can be fixed by add a simple line in SimpleFolderDataset function detecting whether the file is a legit image.

from self-correction-human-parsing.

How do you solve the problem? Can you share you code?

from self-correction-human-parsing.

Hi jasmineytt,

I don't have the code for the moment, but what you can do is to check the image folder and see if there is any mid-product generated by Jupyter Notebook. Those checkpoints would cause problem as they are not images. A simple function would be solely read the image file by checking the file names.

from self-correction-human-parsing.