Skip to content

Commit

Permalink
Updating CITATION and README after NeurIPS acceptance.
Browse files Browse the repository at this point in the history
  • Loading branch information
BorisIvanovic committed Oct 12, 2023
1 parent 5bc8ab5 commit 748b8b1
Show file tree
Hide file tree
Showing 2 changed files with 39 additions and 5 deletions.
22 changes: 19 additions & 3 deletions CITATION.cff
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,23 @@ authors:
given-names: "Boris"
orcid: "https://orcid.org/0000-0002-8698-202X"
title: "trajdata: A unified interface to many trajectory forecasting datasets"
version: 1.0.3
version: 1.3.3
doi: 10.5281/zenodo.6671548
date-released: 2022-06-20
url: "https://github.com/nvr-avg/trajdata"
date-released: 2023-08-22
url: "https://github.com/nvr-avg/trajdata"
preferred-citation:
type: conference-paper
authors:
- family-names: "Ivanovic"
given-names: "Boris"
orcid: "https://orcid.org/0000-0002-8698-202X"
- family-names: "Song"
given-names: "Guanyu"
- family-names: "Gilitschenski"
given-names: "Igor"
- family-names: "Pavone"
given-names: "Marco"
journal: "Proceedings of the Neural Information Processing Systems Track on Datasets and Benchmarks"
month: 12
title: "trajdata: A Unified Interface to Multiple Human Trajectory Datasets"
year: 2023
22 changes: 20 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,11 +1,15 @@
# Unified Trajectory Data Loader
# trajdata: A Unified Interface to Multiple Human Trajectory Datasets

[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
[![Imports: isort](https://img.shields.io/badge/%20imports-isort-%231674b1?style=flat&labelColor=ef8336)](https://pycqa.github.io/isort/)
[![License](https://img.shields.io/badge/License-Apache_2.0-blue.svg)](https://opensource.org/licenses/Apache-2.0)
[![DOI](https://zenodo.org/badge/488789438.svg)](https://zenodo.org/badge/latestdoi/488789438)
[![PyPI version](https://badge.fury.io/py/trajdata.svg)](https://badge.fury.io/py/trajdata)

### Announcements

**Sept 2023**: [Our paper about trajdata](https://arxiv.org/abs/2307.13924) has been accepted to the NeurIPS 2023 Datasets and Benchmarks Track!

## Installation

The easiest way to install trajdata is through PyPI with
Expand Down Expand Up @@ -207,6 +211,20 @@ for t in range(1, sim_scene.scene.length_timesteps):

`examples/sim_example.py` contains a more comprehensive example which initializes a simulation from a scene in the nuScenes mini dataset, steps through it by replaying agents' GT motions, and computes metrics based on scene statistics (e.g., displacement error from the original GT data, velocity/acceleration/jerk histograms).

## Citation

If you use this software, please cite it as follows:
```
@Inproceedings{ivanovic2023trajdata,
author = {Ivanovic, Boris and Song, Guanyu and Gilitschenski, Igor and Pavone, Marco},
title = {{trajdata}: A Unified Interface to Multiple Human Trajectory Datasets},
booktitle = {{Proceedings of the Neural Information Processing Systems (NeurIPS) Track on Datasets and Benchmarks}},
month = dec,
year = {2023},
address = {New Orleans, USA},
url = {https://arxiv.org/abs/2307.13924}
}
```

## TODO
- Create a method like finalize() which writes all the batch information to a TFRecord/WebDataset/some other format which is (very) fast to read from for higher epoch training.
- Add more examples to the README.

0 comments on commit 748b8b1

Please sign in to comment.