oneline-wsq / nuscenes

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Sentence label generation based on nuScenes dataset

This code allows you to generate sentence label based on nuScenes dataset. The concept of the sentence label comes from Tesla's AI day in 2021. In the AI day sharing, they proposed an autoregressive lane language model.

Inspired by this, we decided to build and train a similar model on a publicly available dataset. This repository is focused on the preliminary data preparation and the generation of the labels required for network training.

Data processing flow

  1. Find the map patch corresponding to the current frame according to ego pose;
  2. Intercept the lane + lane connector in the map patch;
  3. Find the key point according to the incoming degree & outcoming degree provided by the connectivity;
  4. Divide the lanes according to length and curvature to determine the control points;
  5. Iterate/search the key points & control points in the map patch, and generate the language of lanes;

The final generated result is shown in the following figure.

About


Languages

Language:Python 100.0%