Skip to content

Code of CVPR2025 Paper 《T2SG: Traffic Topology Scene Graph for Topology Reasoning in Autonomous Driving》

Notifications You must be signed in to change notification settings

MICLAB-BUPT/T2SG

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

T2SG:
Traffic Topology Scene Graph for Topology Reasoning in Autonomous Driving

arXiv
State Key Laboratory of Networking and Switching Technology, Beijing University of Posts and Telecommunications, China

In autonomous driving, understanding a traffic scene isn’t just about detecting lanes—it’s about grasping how lanes connect, why they are structured the way they are, and how traffic signals govern their use. Can current HD mapping methods build a coherent, structured representation that captures this logic? Or do they merely trace geometry without reasoning? We investigate these questions by introducing T2SG (Traffic Topology Scene Graph), a unified scene graph that explicitly models lane topology and signal-guided relationships,and propose TopoFormer, a one-stage transformer that generates such graphs through geometric aggregation and counterfactual structural reasoning. By evaluating on OpenLane-V2, we show not only that T²SG enables more accurate and explainable HD maps, but also that it significantly enhances downstream spatial reasoning for autonomous systems.

Release

  • 2025-03-11 : Our paper T2SG is accepted by CVPR 2025 as Poster! See u in Nashville!

Contents

Results

Evaluation Setups: We report comparative results between our model and the baselines on two tasks: traffic topology scene graph generation and OpenLaneV2.

Getting Started

Dataset

Our experiments are conducted on OpenLane-V2. Please download the dataset from the official OpenLane-V2 release, create a data/ folder under the project root, and place the dataset inside it (e.g., data/OpenLane-V2/).

Prerequisites

  • Linux
  • Python 3.8.20
  • NVIDIA GPU + CUDA 11.1
  • PyTorch 1.9.1+cu111
  • torchvision 0.10.1+cu111
  • mmcv-full 1.5.2, mmdet 2.26.0, mmdet3d 1.0.0rc6
  • openlanev2 1.1.0

Train and Evaluate

Set the environment variables before running the command.

export TORCH_HOME=work_dirs/
export PYTHONPATH=$PYTHONPATH:"./"

Train

Stage 1

./tools/dist_train.sh <WORK_DIR> 8

Stage 2(finetune)

./tools/dist_train_finetune.sh <WORK_DIR> 8

Evaluate

./tools/dist_test.sh <WORK_DIR> 8

Note: <WORK_DIR>: path to save logs and checkpoints (e.g., work_dirs/exp_01)

Acknowledgement

Our code is build upon TopoNet. We acknowledge their team for open-sourcing their implementation.

Citation

If you find T2SG is useful in your research or applications, please consider citing it by the following BibTeX entry.

@InProceedings{Lv_2025_CVPR,
    author    = {Lv, Changsheng and Qi, Mengshi and Liu, Liang and Ma, Huadong},
    title     = {T2SG: Traffic Topology Scene Graph for Topology Reasoning in Autonomous Driving},
    booktitle = {Proceedings of the Computer Vision and Pattern Recognition Conference (CVPR)},
    month     = {June},
    year      = {2025},
    pages     = {17197-17206}
}

About

Code of CVPR2025 Paper 《T2SG: Traffic Topology Scene Graph for Topology Reasoning in Autonomous Driving》

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •