Skip to content

Add support for Waymo Open Dataset v1.3.0 training, validation, and test splits#89

Open
siddhi47 wants to merge 1 commit intowaymo-research:mainfrom
siddhi47:main
Open

Add support for Waymo Open Dataset v1.3.0 training, validation, and test splits#89
siddhi47 wants to merge 1 commit intowaymo-research:mainfrom
siddhi47:main

Conversation

@siddhi47
Copy link
Copy Markdown

@siddhi47 siddhi47 commented Aug 4, 2025

This PR introduces three new DatasetConfig entries to enable loading of the full Waymo Open Dataset Motion v1.3.0 release:

  • Training Set: Includes 1000 sharded TFRecord files with an optional cap of 10,000 range points per example.
  • Validation Set: Includes 150 TFRecord shards.
  • Test Set: Includes 150 TFRecord shards.
WOD_1_3_0_TRAINING = DatasetConfig(
    path="gs://waymo_open_dataset_motion_v_1_3_0/uncompressed/tf_example/training/training_tfexample.tfrecord@1000",
    max_num_rg_points=10000,
    data_format=DataFormat.TFRECORD,
)

WOD_1_3_0_VALIDATION = DatasetConfig(
    path="gs://waymo_open_dataset_motion_v_1_3_0/uncompressed/tf_example/validation/validation_tfexample.tfrecord@150",
    data_format=DataFormat.TFRECORD,
)

WOD_1_3_0_TEST = DatasetConfig(
    path="gs://waymo_open_dataset_motion_v_1_3_0/uncompressed/tf_example/testing/testing_tfexample.tfrecord@150",
    data_format=DataFormat.TFRECORD,
)

Why:

These additions ensure the codebase is compatible with the updated v1.3.0 Waymo Motion dataset release and support full-scale training and evaluation using the latest split structures.

@google-cla
Copy link
Copy Markdown

google-cla bot commented Aug 4, 2025

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant