THU-DA-6D-Pose-Group/CATRE

训练时无法找到数据

Closed this issue · 0 comments

AttributeError: Caught AttributeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop
data = fetcher.fetch(index)
File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/fetch.py", line 49, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/media/yusheng/F/code/CATRE-main/core/catre/../../core/catre/datasets/data_loader.py", line 879, in getitem
processed_data = self.read_data(dataset_dict)
File "/media/yusheng/F/code/CATRE-main/core/catre/../../core/base_data_loader.py", line 93, in read_data
return self.read_data_train(dataset_dict)
File "/media/yusheng/F/code/CATRE-main/core/catre/../../core/catre/datasets/data_loader.py", line 438, in read_data_train
self._get_mug_meta(dataset_name)
File "/media/yusheng/F/code/CATRE-main/core/catre/../../core/catre/datasets/data_loader.py", line 357, in _get_mug_meta
ref_key = dset_meta.ref_key
File "/home/yusheng/detectron2/detectron2/data/catalog.py", line 132, in getattr
f"Attribute '{key}' does not exist in the metadata of dataset '{self.name}': "
AttributeError: Attribute 'ref_key' does not exist in the metadata of dataset 'nocs_train_real': metadata is empty.

请问大佬,测试的时候没有问题,但是在训练的时候就找不到数据,这是什么情况?