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

about the visualization of multiagent #17

Open kronaldo123 opened 2 months ago

kronaldo123 commented 2 months ago

Hello

I am very glad that your team can give such good research and open source code.

We are very interested in your work. I have some small questions about your work: the visualization code in the notebook seems to be applicable to single-agent, because I noticed the directory ./src/model/multiagent/trainer_forecast_ma.py you provided. Have you provide the code for visualizing multiagent?