SuLvXiangXin / zipnerf-pytorch

Unofficial implementation of ZipNeRF
Apache License 2.0
783 stars 85 forks source link

A question about my data #73

Open LoveMYChen opened 11 months ago

LoveMYChen commented 11 months ago

How to slove thie problem? Traceback (most recent call last): File "train.py", line 387, in app.run(main) File "/root/miniconda3/lib/python3.8/site-packages/absl/app.py", line 312, in run _run_main(main, args) File "/root/miniconda3/lib/python3.8/site-packages/absl/app.py", line 258, in _run_main sys.exit(main(argv)) File "train.py", line 73, in main dataset = datasets.load_dataset('train', config.data_dir, config) File "/root/autodl-tmp/zipnerf-pytorch/internal/datasets.py", line 39, in load_dataset return dataset_dict[config.dataset_loader](split, train_dir, config) File "/root/autodl-tmp/zipnerf-pytorch/internal/datasets.py", line 278, in init self._load_renderings(config) File "/root/autodl-tmp/zipnerf-pytorch/internal/datasets.py", line 543, in _load_renderings pose_data = NeRFSceneManager(colmap_dir).process() File "/root/autodl-tmp/zipnerf-pytorch/internal/datasets.py", line 66, in process cam = self.cameras[1] KeyError: 1

Thanks!

yimellia commented 11 months ago

You can try to change cam = self.cameras[1] to cam = self.cameras[2]. If it does not work, you can print(self.cameras) and check the index which comes before the camera object. Then change it to cam = self.cameras[].