autonomousvision / neat

[ICCV'21] NEAT: Neural Attention Fields for End-to-End Autonomous Driving
MIT License
310 stars 47 forks source link

How to get the visualization results as shown in README #4

Closed chhanganivarun closed 1 year ago

chhanganivarun commented 2 years ago

I am not able to replicate the gif shown in README wherein you are able to visualize waypoint offsets too along with dashcam and semantic segmentation from BEV.

Could you please share the scripts for the same?

kashyap7x commented 2 years ago

If you provide a SAVE_PATH while evaluating NEAT, the RGB image, semantics and waypoint offsets all get saved to your disk: https://github.com/autonomousvision/neat/blob/686f4a0b5b5bf20c99f323e9542f5b68808df2de/leaderboard/team_code/neat_agent.py#L266

We created the visualization in the gif using a video editing tool (https://kdenlive.org/en/), and not through a script.