ZikangZhou / QCNet

[CVPR 2023] Query-Centric Trajectory Prediction
https://openaccess.thecvf.com/content/CVPR2023/papers/Zhou_Query-Centric_Trajectory_Prediction_CVPR_2023_paper.pdf
Apache License 2.0
481 stars 76 forks source link

Visualization methods #8

Closed Lionel-Lee closed 1 year ago

Lionel-Lee commented 1 year ago

Hi Zikang,

Cheers for your wonderful work! Is it possible for you guys to release the visualization code as well? As I'm not quite familiar with the argoverse dataset and its toolkit, it would be great if you could share your visualization methods. Thanks!

ZikangZhou commented 1 year ago

Our internal visualization tool is based on the official one in av2 API. You can find some useful utils here: https://github.com/argoverse/av2-api/blob/main/tutorials/generate_forecasting_scenario_visualizations.py, which can produce beautiful videos of traffic scenarios.