This version brings with it a major visualization upgrade, now with HTML-based interactive plots and animations! Additionally, we provide `trajdata.utils.batch_utils.SceneTimeBatcher`, a batch_sampler that can be fed into a standard torch dataloader, for use cases where one wants to loop through a whole Agent-centric dataset, but calculate statistics grouped by individual timesteps in scenes.
See the notes below for more details.
- Users can now create interactive plots via Bokeh's HTML-based visualization library. Beyond static figures, users can also create interactive animations! Take a look at `examples/visualization_example.py` to see how you can use these features too!
- `SceneTimeBatcher` is a batch sampler that can be fed into a standard PyTorch dataloader, e.g.,
dataset = UnifiedDataset(
desired_data=["nusc_mini-mini_train"],
centric="agent"
)
dataloader = DataLoader(
dataset,
batch_sampler=SceneTimeBatcher(dataset),
collate_fn=dataset.get_collate_fn(),
num_workers=4,
)
Each batch from the resulting dataset is an `AgentBatch`, but with each element corresponding to each agent at a particular timestep in a particular scene. An example is provided at `examples/scenetimebatcher_example.py`.
- Added information about the nuPlan dataset to `DATASETS.md`, additional tests related to the above additions, and bugfixes.