search

xupei0610 / ContextVAE

[ICRA 2024]Context-Aware Timewise VAEs for Real-Time Vehicle Trajectory Prediction
MIT License
43 stars 6 forks source link

Thank you for the author's outstanding work. Your paper's visualization is very beautiful. May I ask if you could provide the visualization code? #1

Closed zhoulian-sudo closed 10 months ago

zhoulian-sudo commented 10 months ago

Thank you for the author's outstanding work. Your paper's visualization is very beautiful. May I ask if you could provide the visualization code?Thanks.

xupei0610 commented 10 months ago

The APIs from different datasets differ a lot. For example, nuScenes directly provides the API to get semantic maps, while for Lyft (having an overview map) and Waymo (only having the local map for each scenario), we have to draw road lines by ourselves.

Here is a script that I used to visualize the map of nuScenes. Hope it could be helpful https://gist.github.com/xupei0610/98d99b1b62ced0bce55d181ed331fe89

xupei0610 commented 10 months ago

If you have additional questions about visualization, you can reach me by my email, which you can find from my website https://pei-xu.github.io/.

Thank you.