search

zhengqili / CGIntrinsics

This is the CGIntrinsics implementation described in the paper "CGIntrinsics: Better Intrinsic Image Decomposition through Physically-Based Rendering, Z. Li and N. Snavely, ECCV 2018".
MIT License
75 stars 17 forks source link

changes the train_list_Render to read from intrinsics_final #6

Open jrock08 opened 5 years ago

jrock08 commented 5 years ago

Pretty sure this is the correct behavior?

jrock08 commented 5 years ago

Alternatively, based on a crash/bug I'm seeing when I try to run this, is there a different data source that I should have set up intrinsics/ in the data directory?

 File "train.py", line 149, in <module>
    data_R = next(iterator_Render, None)
  File "/home/jrock/Documents/CGIntrinsics/env/local/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 637, in __next__
    return self._process_next_batch(batch)
  File "/home/jrock/Documents/CGIntrinsics/env/local/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 658, in _process_next_batch
    raise batch.exc_type(batch.exc_msg)
RuntimeError: Traceback (most recent call last):
  File "/home/jrock/Documents/CGIntrinsics/env/local/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 138, in _worker_loop
    samples = collate_fn([dataset[i] for i in batch_indices])
  File "/home/jrock/Documents/CGIntrinsics/env/local/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 229, in default_collate
    return {key: default_collate([d[key] for d in batch]) for key in batch[0]}
  File "/home/jrock/Documents/CGIntrinsics/env/local/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 229, in <dictcomp>
    return {key: default_collate([d[key] for d in batch]) for key in batch[0]}
  File "/home/jrock/Documents/CGIntrinsics/env/local/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 229, in default_collate
    return {key: default_collate([d[key] for d in batch]) for key in batch[0]}
  File "/home/jrock/Documents/CGIntrinsics/env/local/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 229, in <dictcomp>
    return {key: default_collate([d[key] for d in batch]) for key in batch[0]}
  File "/home/jrock/Documents/CGIntrinsics/env/local/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 209, in default_collate
    return torch.stack(batch, 0, out=out)
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 512 and 768 in dimension 2 at /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1333