/talk2bev

Talk2BEV: Language-Enhanced Bird's Eye View Maps (ICRA'24)

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

Talk2BEV: Language-Enhanced Bird's Eye View Maps

Project Page | ArXiv | Video

Vikrant Dewangan* 1, Tushar Choudhary* 1, Shivam Chandhok* 2, Shubham Priyadarshan 1, Anushka Jain 1, Arun K. Singh 3, Siddharth Srivastava 4, Krishna Murthy Jatavallabhula $^\dagger$ 5, K. Madhava Krishna $^\dagger$ 1

1 International Institute of Information Technology Hyderabad, 2 University of British Columbia, 3 University of Tartu 4 TensorTour Inc 5 MIT-CSAIL

*denotes equal contribution, $^\dagger$ denotes equal advising

ICRA 2024

methodology.mp4

Abstract

We introduce Talk2BEV, a large vision- language model (LVLM) interface for bird’s-eye view (BEV) maps commonly used in autonomous driving.

While existing perception systems for autonomous driving scenarios have largely focused on a pre-defined (closed) set of object categories and driving scenarios, Talk2BEV eliminates the need for BEV- specific training, relying instead on performant pre-trained LVLMs. This enables a single system to cater to a variety of autonomous driving tasks encompassing visual and spatial reasoning, predicting the intents of traffic actors, and decision- making based on visual cues.

We extensively evaluate Talk2BEV on a large number of scene understanding tasks that rely on both the ability to interpret freefrom natural language queries, and in grounding these queries to the visual context embedded into the language-enhanced BEV map. To enable further research in LVLMs for autonomous driving scenarios, we develop and release Talk2BEV-Bench, a benchmark encom- passing 1000 human-annotated BEV scenarios, with more than 20,000 questions and ground-truth responses from the NuScenes dataset.

Data Preparation

Please download the NuScenes v1.0-trainval dataset. Our dataset consists of 2 parts - Talk2BEV-Base and Talk2BEV-Captions, consisting of base (crops, perspective images, bev area centroids) and crop captions respectively.

Download links

We provide 2 Links to the Talk2BEV dataset (Talk2BEV-Mini (captions only) and Talk2BEV-Full) are provided below. The dataset is hosted on Google Drive. Please download the dataset and extract the files to the data folder.

Name Base Captions Bench Link
Talk2BEV-Mini link
Talk2BEV-Full TODO

If you want to generate the dataset from scratch, please follow the process here. The format for each of the data parts is described in format.

Evaluation

Evaluation on Talk2BEV happens via 2 methods - MCQs (from Talk2BEV-Bench) and Spatial Operators. We use GPT-4 for our evaluation. Please follow the instructions in GPT-4 and initialize the API key and Organization in your os env.

ORGANIZATION=<your-organization>
API_KEY=<your-api-key>

Evaluating - MCQs

To obtain the accuracy for a MCQs, please run the following command:

cd evaluation
python eval_mcq.py

This will yield the accuracy for the MCQs.

Evaluating Spatial Operators

To obtain the distance error, IoU for a MCQs, please run the following command:

cd evaluation
python eval_spops.py

Click2Chat

We also allow free-form conversation with the BEV. Please follow the instructions in Click2Chat to chat with the BEV.

Talk2BEV-Bench

TO BE RELEASED

👉 TODO

  • Spatial operators evaluation pipeline
  • Add links to BEV crops -- Release Talk2BEV-Full
  • Release Talk2BEV-Bench