AttributeError: 'NoneType' object has no attribute 'shape'

[TimLeeCV]

Epoch 0: 17%|█▉ | 247/1434 [03:11<15:21, 1.29it/s, loss=13.3, v_num=3]Traceback (most recent call last): File "/home/lee/code/GeoFormer/lightning/train_homo_geoformer.py", line 134, in main() File "/home/lee/code/GeoFormer/lightning/train_homo_geoformer.py", line 130, in main trainer.fit(model, datamodule=data_module) File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 458, in fit self._run(model) File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 756, in _run self.dispatch() File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 797, in dispatch self.accelerator.start_training(self) File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/pytorch_lightning/accelerators/accelerator.py", line 96, in start_training self.training_type_plugin.start_training(trainer) File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/pytorch_lightning/plugins/training_type/training_type_plugin.py", line 144, in start_training self._results = trainer.run_stage() File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 807, in run_stage return self.run_train() File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 869, in run_train self.train_loop.run_training_epoch() File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/pytorch_lightning/trainer/training_loop.py", line 491, in run_training_epoch for batch_idx, (batch, is_last_batch) in train_dataloader: File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/pytorch_lightning/profiler/profilers.py", line 112, in profile_iterable value = next(iterator) File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/pytorch_lightning/trainer/supporters.py", line 534, in prefetch_iterator for val in it: File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/pytorch_lightning/trainer/supporters.py", line 464, in next return self.request_next_batch(self.loader_iters) File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/pytorch_lightning/trainer/supporters.py", line 478, in request_next_batch return apply_to_collection(loader_iters, Iterator, next) File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/pytorch_lightning/utilities/apply_func.py", line 85, in apply_to_collection return function(data, *args, **kwargs) File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 517, in next data = self._next_data() File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 557, in _next_data data = self._dataset_fetcher.fetch(index) # may raise StopIteration File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/lee/.conda/envs/geoformer/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/lee/code/GeoFormer/homodataset/HomoDataset.py", line 162, in getitem image0, image1, homography, valid_mask_left, valid_mask_right, is_neg = self.get_images_labels(index) File "/home/lee/code/GeoFormer/homodataset/HomoDataset.py", line 130, in get_images_labels image0, image1, homography = self.get_pair(file_path) File "/home/lee/code/GeoFormer/homodataset/HomoDataset.py", line 87, in get_pair height, width = image.shape[0:2] AttributeError: 'NoneType' object has no attribute 'shape' Epoch 0: 17%|█▋ | 247/1434 [03:12<15:22, 1.29it/s, loss=13.3, v_num=3]