Emerge-Lab / gpudrive

GPU-acceleration of Nocturne via Madrona
https://arxiv.org/abs/2408.01584
MIT License
232 stars 20 forks source link

Support for re-building the WOMD dataset #260

Closed daphnecor closed 1 month ago

daphnecor commented 1 month ago

Description

This PR supports the re-building of the JSON files from the Waymo Open Motion Dataset.

Note: This PR is co-authored by @nadarenator

Todo's

Todo's

nadarenator commented 1 month ago

Things to address:

daphnecor commented 1 month ago
  • Will have to downgrade to older version of waymo-open-dataset due to incompatibility with python 3.11. Unless significant changes in protobuf this should be okay, else we will have use a separate python env just for re-building.

Either we can do that, or we change to Python 3.10 (which is what I've been doing so far, see changes in the readme)