Skip to content

Add clipgt dataset support#61

Merged
BorisIvanovic merged 15 commits into
NVlabs:alpasimfrom
maximilianigl:add-mads-utils
Mar 13, 2026
Merged

Add clipgt dataset support#61
BorisIvanovic merged 15 commits into
NVlabs:alpasimfrom
maximilianigl:add-mads-utils

Conversation

@maximilianigl

Copy link
Copy Markdown

Add clipgt support.

Add the MADS (Multi-Agent Driving Simulator) dataset module, which
provides:
- MADSDataset: full dataset integration for loading MADS clipgt data
  including ego and obstacle agent parsing with cubic interpolation,
  overlap filtering, and map caching
- populate_vector_map: constructs VectorMap instances from MADS parquet
  map data (lanes, road edges, traffic signs, wait lines)
- MadsSceneRecord and env_utils registration for dataset discovery
@maximilianigl maximilianigl changed the title Add MADS dataset support Add clipgt dataset support Mar 13, 2026

@BorisIvanovic BorisIvanovic left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you very much, @maximilianigl !

Comment thread src/trajdata/dataset_specific/mads/mads_utils.py Outdated
Comment thread src/trajdata/dataset_specific/mads/mads_utils.py Outdated
Comment thread src/trajdata/utils/env_utils.py Outdated
Comment thread src/trajdata/dataset_specific/mads/mads_utils.py Outdated
Comment thread src/trajdata/dataset_specific/mads/mads_utils.py Outdated
Comment thread src/trajdata/dataset_specific/mads/mads_dataset.py Outdated
Comment thread src/trajdata/dataset_specific/mads/mads_dataset.py Outdated
Comment thread src/trajdata/dataset_specific/mads/mads_dataset.py Outdated
Comment thread src/trajdata/dataset_specific/mads/mads_dataset.py Outdated
Comment thread src/trajdata/dataset_specific/mads/mads_dataset.py Outdated
maximilianigl and others added 11 commits March 13, 2026 19:48
Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com>
Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com>
Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com>
Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com>
Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com>
Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com>
Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com>
Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com>
Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com>
Comment thread src/trajdata/dataset_specific/mads/mads_utils.py Outdated
Comment thread src/trajdata/utils/env_utils.py

@BorisIvanovic BorisIvanovic left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you, @maximilianigl ! Looking good to go, just two minor points left and then I'll merge it

@BorisIvanovic BorisIvanovic merged commit d428ba2 into NVlabs:alpasim Mar 13, 2026
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants