jchengai / forecast-mae

[ICCV'2023] Forecast-MAE: Self-supervised Pre-training for Motion Forecasting with Masked Autoencoders
https://arxiv.org/pdf/2308.09882.pdf
154 stars 16 forks source link

Av2 Multi Agents Forecasting #2

Closed Ross-Stein closed 11 months ago

Ross-Stein commented 1 year ago

Hi @jchengai ,

Firstly, we'd like to express our gratitude for your outstanding work!

We are also interested in the field of multi-agent forecasting. Do you have any plans to release the training and testing/submission code for multi-agent forecasting (especially on the av2 dataset) in the future? Access to the code would be also precious for us.

Could you please let us know if you have any thoughts or a potential timeline for releasing the code? We're eager to further explore your work.

Thanks for your time and contributions. :)

Best regards, Ross

jchengai commented 1 year ago

Hi Ross, thank you for your interest in our work. The multi-agent version incorporates several modifications compared to the current one:

Yes, I would like to provide an exemplary pipeline for multi-agent forecasting (on AV2) if it is beneficial to the community. While I cannot commit to a specific timeline, there is a high likelihood that it will be available within this month :).

Ross-Stein commented 1 year ago

Hi Jie,

thank you for the detailed explanation about the modifications of the multi-agent version!

Also, very happy to hear that you are about to share the code, It would be another great contribution. :) Looking forward to accessing it, thanks!

Best, Ross

jchengai commented 11 months ago

Hi, @Ross-Stein, a multiagent baseline model is now released (performance and checkpoints will be updated once I finish the training). I hope it helps :).

p.s. please note that it's a close one, but not the leaderboard version.

Ross-Stein commented 11 months ago

Hi Jie @jchengai ,

Thanks for your baseline model, It is incredibly useful! :) We're eagerly looking forward to starting the training process.

Warm regards, Ross