训练mmdet3d模型时发生错误
Original Traceback (most recent call last):
File "/HOME/scw6d49/.conda/envs/testvim/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop
data = fetcher.fetch(index)
File "/HOME/scw6d49/.conda/envs/testvim/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/HOME/scw6d49/.conda/envs/testvim/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in <listcomp>
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/data/run01/scw6d49/Project/Testvim/mmdet3d/datasets/nuscenes_dataset.py", line 188, in __getitem__
data = self.prepare_train_data(idx)
File "/data/run01/scw6d49/Project/Testvim/mmdet3d/datasets/custom_3d.py", line 155, in prepare_train_data
~(example['gt_labels_3d']._data != -1).any()):
KeyError: 'gt_labels_3d'
错误代码
def prepare_train_data(self, index):
"""Training data preparation.
Args:
index (int): Index for accessing the target data.
Returns:
dict: Training data dict of the corresponding index.
"""
input_dict = self.get_data_info(index)
if input_dict is None:
return None
self.pre_pipeline(input_dict)
example = self.pipeline(input_dict)
if self.filter_empty_gt and \
(example is None or
~(example['gt_labels_3d']._data != -1).any()):
return None
return example
