search

Tsinghua-MARS-Lab / ViP3D

MIT License
133 stars 19 forks source link

How can I get the nuscenes_prediction_infos_val.json? #11

Closed sean-wade closed 10 months ago

sean-wade commented 10 months ago

I want to evaluate on nuscenes-mini dataset. But the nuscenes_prediction_infos_val.json in the project is the whole dataset's?

Is there any code that generate nuscenes_prediction_infos_val.json?

GentleSmile commented 10 months ago

Generation code has been uploaded now. First rename your data/nuscenes-mini to data/nuscenes, then generate nuscenes prediction infos at test time:

PYTHONPATH=. python tools/test.py plugin/vip3d/configs/vip3d_resnet50_3frame.py work_dirs/vip3d_resnet50_3frame.1/epoch_24.pth --eval bbox --cfg-options data.test.generate_nuscenes_prediction_infos_val=True data.workers_per_gpu=0

Then it will be saved at data/nuscenes.

sean-wade commented 10 months ago

OK, thanks for the reply!