KeyError: 'CAM_FRONT'

[TheRunningHorse]

I generated data/nuscenes/nuscenes_infos_train.pkl and data/nuscenes/nuscenes_infos_val.pkl through python tools/create_data.py nuscenes --root-path ./data/nuscenes --out-dir ./data/nuscenes --version v1.0-mini --extra-tag
and tested the pretrained model using the cmd below:
python tools/test.py projects/configs/CAPE-T/capet_VoV99_800x320_24ep_wocbgs_load_dd3d_pretrain.py ckpts/capet_vov99_800x320_epoch_24.pth --eval bbox
There is an error
projects.mmdet3d_plugin
load checkpoint from local path: ckpts/capet_vov99_800x320_epoch_24.pth
The model and loaded state dict do not match exactly

unexpected key in source state_dict: pts_bbox_head.transformer.image_plane, pts_bbox_head.transformer.bound, pts_bbox_head.transformer.cam_bound

[ ] 0/81, elapsed: 0s, ETA:
[ ] 1/81, 1.8 task/s, elapsed: 1s, ETA: 45sTraceback (most recent call last):
File "/home/msy/spatial-modeling/CAPE/tools/test.py", line 250, in
main()
File "/home/msy/spatial-modeling/CAPE/tools/test.py", line 220, in main
outputs = single_gpu_test(model, data_loader, args.show, args.show_dir)
File "/home/msy/spatial-modeling/CAPE/deps/mmdetection3d/mmdet3d/apis/test.py", line 37, in single_gpu_test
for i, data in enumerate(data_loader):
File "/home/msy/anaconda3/envs/CAPE_new/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 521, in next
data = self._next_data()
File "/home/msy/anaconda3/envs/CAPE_new/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1203, in _next_data
return self._process_data(data)
File "/home/msy/anaconda3/envs/CAPE_new/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1229, in _process_data
data.reraise()
File "/home/msy/anaconda3/envs/CAPE_new/lib/python3.8/site-packages/torch/_utils.py", line 425, in reraise
raise self.exc_type(msg)
KeyError: Caught KeyError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/home/msy/anaconda3/envs/CAPE_new/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop
data = fetcher.fetch(index)
File "/home/msy/anaconda3/envs/CAPE_new/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/msy/anaconda3/envs/CAPE_new/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/msy/spatial-modeling/CAPE/deps/mmdetection3d/mmdet3d/datasets/custom_3d.py", line 355, in getitem
return self.prepare_test_data(idx)
File "/home/msy/spatial-modeling/CAPE/deps/mmdetection3d/mmdet3d/datasets/custom_3d.py", line 172, in prepare_test_data
example = self.pipeline(input_dict)
File "/home/msy/anaconda3/envs/CAPE_new/lib/python3.8/site-packages/mmdet/datasets/pipelines/compose.py", line 41, in call
data = t(data)
File "/home/msy/spatial-modeling/CAPE/./projects/mmdet3d_plugin/datasets/pipelines/loading.py", line 125, in call
results['filename'].extend([sweep[sensor]['data_path'] for sensor in self.sensors])
File "/home/msy/spatial-modeling/CAPE/./projects/mmdet3d_plugin/datasets/pipelines/loading.py", line 125, in
results['filename'].extend([sweep[sensor]['data_path'] for sensor in self.sensors])
KeyError: 'CAM_FRONT'

[xudh777]

Sorry to bother you, I also encountered the same problem. Has it been resolved?
The wrong location is： /projects/mmdet3d_plugin/datasets/pipelines/loading.py 121
The reason for the error is：

sweep: dict_keys(['data_path', 'type', 'sample_data_token', 'sensor2ego_translation', 'sensor2ego_rotation', 'ego2global_translation', 'ego2global_rotation', 'timestamp', 'sensor2lidar_rotation', 'sensor2lidar_translation'])