search

idsc-frazzoli / waymax

A JAX-based simulator for autonomous driving research.
Other
0 stars 0 forks source link

faster_rendering #10

Closed nicolaloi closed 5 days ago

nicolaloi commented 6 days ago

Call ax.plot() once instead of twice in plot_numpy_bounding_boxes, I got a 40% faster visualization.plot_simulator_state (tested before the introduction of plot_numpy_rays)

alezana commented 6 days ago

cool :) could you annotate in the code that boxes and arrows are plotted together? have you tested it in multi-agent setups? maybe @jubilantrou could test it

nicolaloi commented 6 days ago

I don't know how to start a multi-agent env.

jubilantrou commented 5 days ago

I've tested it in the waymax env. Each evaluation process can save around 1 min.