MCZhi / GameFormer-Planner

[ICCV & CVPR Workshop] Learning-enabled Interactive Prediction and Planning Framework for Autonomous Vehicles
https://mczhi.github.io/GameFormer/
MIT License
143 stars 17 forks source link

training data #4

Closed ynw2021 closed 11 months ago

ynw2021 commented 11 months ago

Hi huang,

Thank you for sharing the great work! When I check the data_process.py, it seems that you only used the test 14 scenario. Is that the same thing you did when training your nuplan challenge submittion? Could you give more details about which subset you used for training, and why not to use the whole dataset?

Bests,

MCZhi commented 11 months ago

Hi, @ynw2021. We used all the accessible data for the nuPlan challenge. In this open-source version, we have only used the selected testing scenario types for quicker data processing. Feel free to remove this restriction from your code if your want to further enhance the model perfermance.

ynw2021 commented 11 months ago

Hi huang, I find out that it will cost thousands of hours to preprocess the whole dataset. Is this normal or did you limit the number of scenario per type for the challenge?

MCZhi commented 11 months ago

Hi, @ynw2021. The processing of the entire dataset can take a significant amount of time, which is normal. We initially tried to speed up the process by using multiprocessing, but it had the opposite effect and slowed it down, possibly due to problems with querying the SQL dataset. We did the processing on a cluster computing server, so it did not take that long (thousands of hours) but still hundreds of hours.

ynw2021 commented 11 months ago

Thanks a lot for the help. Another question: Is this warning ok when processing data?image

MCZhi commented 11 months ago

Yes, it is OK and doesn't affect anything.

ynw2021 commented 10 months ago

Another thing,did you set the timestamp_threshold_s?