crownconv360depth icon indicating copy to clipboard operation
crownconv360depth copied to clipboard

Issues with importing test dataset

Open LyceanEM opened this issue 2 years ago • 0 comments

I downloaded the linked datasets to try out the model, and I get the following error

TypeError: Caught TypeError in DataLoader worker process 0. Original Traceback (most recent call last): File "/home/timtitan/anaconda3/envs/pytorch3d/lib/python3.9/site-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop data = fetcher.fetch(index) File "/home/timtitan/anaconda3/envs/pytorch3d/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/timtitan/anaconda3/envs/pytorch3d/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 49, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/timtitan/Documents/10-19-Research-Projects/16-Spatial Fingerprinting/16.02 Spatial Mapping/crownconvolution/crownconv360depth/dataloader/icosahedron_dataset.py", line 82, in getitem sample = self.root_dataset[idx] File "/home/timtitan/Documents/10-19-Research-Projects/16-Spatial Fingerprinting/16.02 Spatial Mapping/crownconvolution/crownconv360depth/dataloader/omnistereo_dataset.py", line 59, in getitem sample['idepth'] = load_invdepth(depth_path) File "/home/timtitan/Documents/10-19-Research-Projects/16-Spatial Fingerprinting/16.02 Spatial Mapping/crownconvolution/crownconv360depth/dataloader/omnistereo_dataset.py", line 76, in load_invdepth invdepth = (invd_value / 100.0) / (min_depth * 655) + np.finfo(np.float32).eps TypeError: unsupported operand type(s) for /: 'NoneType' and 'float'

This happens both when I try to train the model fresh, or when I try using the downloadable checkpoints. It seems like there is an import issue when the model checks for invd_value = cv2.imread(filename, cv2.IMREAD_ANYDEPTH) and for some files it returns None.

LyceanEM avatar Apr 26 '22 10:04 LyceanEM