Repository navigation

Commit d428ba2

and

authored

Add clipgt dataset support (#61)

* Add MADS dataset support Add the MADS 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 * Fix test * Update src/trajdata/dataset_specific/mads/mads_utils.py Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com> * Update src/trajdata/dataset_specific/mads/mads_utils.py Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com> * Update src/trajdata/utils/env_utils.py Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com> * Update src/trajdata/dataset_specific/mads/mads_utils.py Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com> * Update src/trajdata/dataset_specific/mads/mads_utils.py Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com> * Update src/trajdata/dataset_specific/mads/mads_dataset.py Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com> * Update src/trajdata/dataset_specific/mads/mads_dataset.py Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com> * Update src/trajdata/dataset_specific/mads/mads_dataset.py Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com> * Update src/trajdata/dataset_specific/mads/mads_dataset.py Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com> * Update mads_dataset.py * fixes * Update env_utils.py * Update mads_utils.py --------- Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com>

1 parent c4c3c6d commit d428ba2Copy full SHA for d428ba2

6 files changed

src/trajdata
- dataset_specific
  - mads
  - scene_records.py
- utils
  - env_utils.py
tests
- test_state.py

`‎src/trajdata/dataset_specific/mads/init.py‎`

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+from .mads_dataset import MADSDataset`

Comments

(0)