danijar / dreamerv3

Mastering Diverse Domains through World Models
https://danijar.com/dreamerv3
MIT License
1.28k stars 219 forks source link

Applied to autonomous driving #105

Open StephenGordan opened 7 months ago

StephenGordan commented 7 months ago

Can DreamerV3 be used in self-driving cars? Are there any related works for reference?

Hamptonjc commented 7 months ago

Somewhat related: https://arxiv.org/abs/2309.17080

AFAIK Comma.ai is also exploring the use of world models to train a driving policy.

gaodechen commented 3 months ago

Yes!

Feel free to check out our recent project and demos at https://github.com/ucd-dare/CarDreamer and there is a technical report for more details

Dreamer is a fantastic toolbox to explore RL for self-driving. We recently made some training and evaluation tasks, such as right turn, left turn, roundabout, overtake, for self-driving in CARLA. It may take only 1-4 hours for DreamerV3 to master each task which was quite amazing considering it learns the policy from scratch in imagination.

- Check out demos from bird's eye views

Roundabout Right Turn Overtake
roundabout right_turn_hard overtake

- CARLA supports highly realistic rendering and sensors like cameras, LiDARs

Left Turn Hard Camera Right Turn Hard Camera Roundabout Camera
left_turn_hard_camera right_turn_hard_camera roundabout_camera
Right Turn Hard with Shared Intentions Right Turn Hard w/o Sharing Intentions
right_turn_hard right_turn_hard_no_wpt