~1500% faster video generation (#15)

* ~1500% faster video generation - to solve in a future PR: - when video_path = None with blit enabled, axis ticks are not correctly updated * Update waymax/visualization/gokart_viz.py Co-authored-by: code-review-doctor[bot] <72320148+code-review-doctor[bot]@users.noreply.github.com> * Update waymax/visualization/gokart_viz.py Co-authored-by: code-review-doctor[bot] <72320148+code-review-doctor[bot]@users.noreply.github.com> * default backend --------- Co-authored-by: code-review-doctor[bot] <72320148+code-review-doctor[bot]@users.noreply.github.com>
idsc-frazzoli · Jan 9, 2025 · 04ee2e4 · 04ee2e4
1 parent d65c841
commit 04ee2e4
Show file tree

Hide file tree

Showing 2 changed files with 599 additions and 0 deletions.
diff --git a/waymax/visualization/__init__.py b/waymax/visualization/__init__.py
@@ -20,3 +20,4 @@
 from waymax.visualization.viz import plot_single_agent_brax_timestep
 from waymax.visualization.viz import plot_traffic_light_signals_as_points
 from waymax.visualization.viz import plot_trajectory
+from waymax.visualization.gokart_viz import create_video_simulator_state