Skip to content

Tags: NVlabs/trajdata

Tags

v1.4.0

Toggle v1.4.0's commit message
Adding quality of life improvements, map area querying, fixing nuPlan…

… lane ID format, and the ability to cache the data index.

v1.3.3

Toggle v1.3.3's commit message
Adding nuPlan info since we now support the full nuPlan dataset!

v1.3.2

Toggle v1.3.2's commit message
Adding support for the INTERACTION dataset and SDD pedestrian dataset!

v1.3.1

Toggle v1.3.1's commit message

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
Waymo Dataset Support (#21)

* Install waymo devkit

* Create dataset_specific folders for waymo dataset

* Implement functions for waymo

* More method implementations

* Format agent data

* Drop agents with only one detection for waymo

* More waymo map implementation

* Refactor and add parallelization to waymo data loading

* Add name check for waymo dataset

* Refactor cache_map

* Implement traffic light status extraction from waymo

* Revert accidental change

* Add tqdm to extract_vectorized function

* Iterate to get the max/min point in waymo map

* Set origin shifting to be 0

* Remove shallow copy

* Update

* Remove commented code and add linear interpolation

* Refactor waymo data loading

* Refactor waymo_dataset implementation

* Fix major bugs

* More debugging

* Add interpolation and remove hard-coded value

* Fix interpolation and first/last timesteps

* Fix interpolation

* Update agent data extraction

* Fix first/last_timestep

* remove unnecessary lines

* Lots of quality of life improvements in preparation for merging (formatting, a bit of refactoring and speedups, parallelism for maps, etc). Remaining big todo is fixing map element boundaries...

* Small formatting post-merge.

* Initial solution for Waymo boundaries.

* Saving work quickly to try something else.

* We'll take it. Just remaining is the connectivity fix and then we can call it.

* Traffic light saving done.

* Small comment fix.

* Fixing lane connectivity, will evaluate it later with improved visualizations.

* Removing TODO comment.

* Lessened the number of hops in the map_api_example.

* Updating version, README, and DATASETS with Waymo info.

* Adding newline.

---------

Co-authored-by: songgua7 <guanyu@loricacyber.com>
Co-authored-by: Boris Ivanovic <bivanovic@nvidia.com>
Co-authored-by: Boris Ivanovic <8534290+BorisIvanovic@users.noreply.github.com>

v1.3.0

Toggle v1.3.0's commit message
Interactive visualizations with Bokeh (+ Animations), the SceneTimeBa…

…tcher, and bugfixes.

v1.2.1

Toggle v1.2.1's commit message
Merge branch 'main' of https://github.com/nvr-avg/trajdata into main

v1.2.0

Toggle v1.2.0's commit message
Adding tensor/array classes with state information, visualization imp…

…rovements, and more for v1.2.0!

v1.1.1

Toggle v1.1.1's commit message
Updating minor version for this small release.

v1.1.0

Toggle v1.1.0's commit message
New vector map API! Lots of changes related to that, starting to supp…

…ort nuPlan, code reorganization, and general bugfixes.

v1.0.8

Toggle v1.0.8's commit message
Adding a few new features like specifying the max number of agents an…

…d if only the ego should be the focus agent, as well as a hotfix bringing back rasterized maps in their original format.