search

open-mmlab / mmdetection3d

OpenMMLab's next-generation platform for general 3D object detection.
https://mmdetection3d.readthedocs.io/en/latest/
Apache License 2.0
5.26k stars 1.53k forks source link

training error #2421

Open guigui6666 opened 1 year ago

guigui6666 commented 1 year ago

hi,i have got some questions on training. after preprocessing nuscenes datasets(mini), I try to train with an existed config with the order: python tools/train.py configs/centerpoint/my_centerpoint_01voxel_second_secfpn_4x8_cyclic_20e_nus.py

and then It will return an error that" KeyError: "NuScenesDataset: 'infos'" "

It seems like we can't find the key value "infos" in nuscenes_dbinfos_train.pkl ? I don't know how to fix it,

Error traceback the error is : Traceback (most recent call last): File "D:\anaconda3\envs\mmdetection3d\lib\site-packages\mmcv\utils\registry.py", line 69, in build_from_cfg return obj_cls(**args) File "e:\pythonproject\mmdetection3d-master\mmdet3d\datasets\nuscenes_dataset.py", line 139, in init test_mode=test_mode) File "e:\pythonproject\mmdetection3d-master\mmdet3d\datasets\custom_3d.py", line 89, in init File "e:\pythonproject\mmdetection3d-master\mmdet3d\datasets\builder.py", line 40, in build_dataset dataset = CBGSDataset(build_dataset(cfg['dataset'], default_args)) File "e:\pythonproject\mmdetection3d-master\mmdet3d\datasets\builder.py", line 44, in build_dataset dataset = build_from_cfg(cfg, DATASETS, default_args) File "D:\anaconda3\envs\mmdetection3d\lib\site-packages\mmcv\utils\registry.py", line 72, in build_from_cfg raise type(e)(f'{obj_cls.name}: {e}') KeyError: "NuScenesDataset: 'infos'"

sunjiahao1999 commented 1 year ago

Check your keys in nuscenes_train.pkl and nuscenes_dbinfos_train.pkl and make sure your pkl file is generated correctly as shown in the tutorial.