Adding quality of life improvements, map area querying, fixing nuPlan…

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

Adding nuPlan info since we now support the full nuPlan dataset!

Adding support for the INTERACTION dataset and SDD pedestrian dataset!

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 <[email protected]>
Co-authored-by: Boris Ivanovic <[email protected]>
Co-authored-by: Boris Ivanovic <[email protected]>

Interactive visualizations with Bokeh (+ Animations), the SceneTimeBa…

…tcher, and bugfixes.

Merge branch 'main' of https://github.com/nvr-avg/trajdata into main

Adding tensor/array classes with state information, visualization imp…

…rovements, and more for v1.2.0!

Updating minor version for this small release.

New vector map API! Lots of changes related to that, starting to supp…

…ort nuPlan, code reorganization, and general bugfixes.

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.