search

swc-17 / SparseDrive

SparseDrive: End-to-End Autonomous Driving via Sparse Scene Representation
MIT License
373 stars 45 forks source link

有关模型test时候，sparsedrive_small_stage2.py的设置问题 #52

Closed guoyage closed 1 month ago

guoyage commented 1 month ago

首先，非常感谢您的精彩工作。在模型的test过程中，我遇到了一些问题。首先，我疑惑在执行test.py命令时，sparsedrive_small_stage2.py中的代码（图片中红色圈出的部分）是否修改为“nuscenes_infos_test.pkl”？？？然而，我将此部分代码修改后，在执行test.py代码时，报如下错误： File "/root/pycharm_project_617/projects/mmdet3d_plugin/datasets/nuscenes_3d_dataset.py", line 310, in get_data_info ego_status=info['ego_status'].astype(np.float32), KeyError: 'ego_status' 这个错误好像是nuscenes_infos_test.pkl文件内的数据内容引起的。我想请问，这可能是什么原因造成的？ 捕获

swc-17 commented 1 month ago

不需要修改，test.pkl对应的是没有标注的test set，用于提交nusc官方榜单，我们这里只能在validation set上做评测。