hhqweasd / G2R-ShadowNet

CVPR2021 From Shadow Generation to Shadow Removal

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ValueError: num_samples should be a positive integer value, but got num_samples=0

subhrajyotidasgupta opened this issue · comments

Hey.

I'm getting a ValueError:
ValueError: num_samples should be a positive integer value, but got num_samples=0
occurring at:

File "train.py", line 95, in <module>
    dataloader = DataLoader(ImageDataset(opt.dataroot, unaligned=True), batch_size=opt.batchSize, shuffle=True, num_workers=opt.n_cpu)

I downloaded the adjusted ISTD dataset from the provided link and providing the root path in the train.py file. What exactly is the issue? Am I missing something?

commented

Please make sure your~/PAISTD8/ folder look like this

PAISTD8
├── train/
    ├── train_A/
    │   └── 90-1.png
    │   └── ...
    ├── train_B/
    │   └── ...
    └── ...

Thanks for adding the dir structure.