search

PJLab-ADG / neuralsim

neuralsim: 3D surface reconstruction and simulation based on 3D neural rendering.
MIT License
582 stars 31 forks source link

COLMAP Dataset no img_names #25

Open awarebayes opened 11 months ago

awarebayes commented 11 months ago

https://github.com/PJLab-ADG/neuralsim/blob/main/dataio/colmap/colmap_dataset.py#L150

  File "/snap/pycharm-professional/348/plugins/python/helpers/pydev/pydevd.py", line 1500, in _exec
    pydev_imports.execfile(file, globals, locals)  # execute the script
  File "/snap/pycharm-professional/348/plugins/python/helpers/pydev/_pydev_imps/_pydev_execfile.py", line 18, in execfile
    exec(compile(contents+"\n", file, 'exec'), glob, loc)
  File "/home/mscherbina/Documents/github_repos/neuralsim/code_single/tools/train.py", line 1322, in <module>
    main_function(bc.parse(print_config=False))
  File "/home/mscherbina/Documents/github_repos/neuralsim/code_single/tools/train.py", line 963, in main_function
    scene_dataloader_train = SceneDataLoader(scene_bank, dataset_impl, config=args.training.dataloader, device=device, is_master=is_master())
  File "/home/mscherbina/Documents/github_repos/neuralsim/dataio/dataloader/base.py", line 86, in __init__
    self._build_cache(cache_device)
  File "/home/mscherbina/Documents/github_repos/neuralsim/dataio/dataloader/base.py", line 168, in _build_cache
    depths.append(self._get_mono_depth(scene_id=scene_id, cam_id=cam_id, frame_ind=frame_ind, device=cache_device).to(torch.float))
  File "/home/mscherbina/Documents/github_repos/neuralsim/dataio/dataloader/base.py", line 362, in _get_mono_depth
    rgb_mono_depth_np = self.dataset_impl.get_mono_depth(scene_id, cam_id, frame_ind + data_frame_offset)
  File "/home/mscherbina/Documents/github_repos/neuralsim/dataio/colmap/colmap_dataset.py", line 142, in get_mono_depth
    img_name = self.img_names[frame_index]
AttributeError: 'COLMAPDataset' object has no attribute 'img_names'