lingtengqiu / Open-PIFuhd

This is an implementation of PIFuHD based on PyTorch
MIT License
349 stars 39 forks source link

runtime error #22

Open gjxwwww8888 opened 1 year ago

gjxwwww8888 commented 1 year ago

Traceback (most recent call last): File "./tools/train_pifu.py", line 56, in train_dataloader = build_dataloader(train_data_set,cfg,args) File "/home/gaojunxiao/train_data/Open-PIFuhd/./utils/dataloader.py", line 33, in build_dataloader return get_loader(dataset,batch_size = batch_size,num_workers = num_workers,collect_fn = collect_fn,shuffle=shuffle,worker_init_function=worker_init_function) File "/home/gaojunxiao/train_data/Open-PIFuhd/./utils/dataloader.py", line 50, in get_loader return torch.utils.data.DataLoader( File "/home/gaojunxiao/anaconda3/envs/pifu/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 266, in init sampler = RandomSampler(dataset, generator=generator) # type: ignore File "/home/gaojunxiao/anaconda3/envs/pifu/lib/python3.8/site-packages/torch/utils/data/sampler.py", line 103, in init raise ValueError("num_samples should be a positive integer " ValueError: num_samples should be a positive integer value, but got num_samples=0

What exactly is the cause of this error? The dataset looks like this,your help would be appreciated. Thank you.

image