crownconv360depth
crownconv360depth copied to clipboard
Issues with importing test dataset
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
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.