RealGen: Retrieval Augmented Generation
for Controllable Traffic Scenarios

Wenhao Ding*1, 2    Yulong Cao*2    Ding Zhao1    Chaowei Xiao2, 3    Marco Pavone2, 4   
1CMU       2NVIDIA       3UW Madison       4Stanford
Published at ECCV 2024 as an Oral paper.

Conventional methods make the model memorize the data distribution for generating. In contrast, our method employs a retriever to query datasets (including external data obtained after training) and uses a generative model to generate scenarios by integrating the information from the retrieved scenarios.

Image

Querying similar and dissimilar scenarios with RealGen

A vehicle (cyan color) goes straight alone on the road.

Image

Query Scenario

Image

Similar Scenario 1

Image

Similar Scenario 2

Image

Dissimilar Scenario

A vehicle (cyan color) yelids other vehicles.

Image

Query Scenario

Image

Similar Scenario 1

Image

Similar Scenario 2

Image

Dissimilar Scenario

Two vehicles drive in opposite directions.

Image

Query Scenario

Image

Similar Scenario 1

Image

Similar Scenario 2

Image

Dissimilar Scenario

Vehicles start to move in an intersection.

Image

Query Scenario

Image

Similar Scenario 1

Image

Similar Scenario 2

Image

Dissimilar Scenario

A vehicle yields another vehicle in an intersection.

Image

Query Scenario

Image

Similar Scenario 1

Image

Similar Scenario 2

Image

Dissimilar Scenario

Generated scenarios retrieved by tags

Yielding

Image
Image

Initial pose and map

Image
Image

Generated scenario

Overtake

Image
Image

Initial pose and map

Image
Image

Generated scenario

U-Turn

Image
Image

Initial pose and map

Image
Image

Generated scenario

Crash scenarios generated by RealGen

Image

Initial pose and map

Image

Generated scenario

Image

Initial pose and map

Image

Generated scenario

Image

Initial pose and map

Image

Generated scenario

Image

Initial pose and map

Image

Generated scenario

Image

Initial pose and map

Image

Generated scenario

Image

Initial pose and map

Image

Generated scenario