8000 Waymo Dataset Support by songguanyu · Pull Request #21 · NVlabs/trajdata · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

Waymo Dataset Support #21

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
merged 45 commits into from
Apr 22, 2023
Merged

Waymo Dataset Support #21

merged 45 commits into from
Apr 22, 2023

Conversation

songguanyu
Copy link
Contributor

No description provided.

songgua7 and others added 30 commits October 23, 2022 18:10
Install waymo devkit
e790218
Create dataset_specific folders for waymo dataset
ad24174
Implement functions for waymo
66d4686
More method implementations
3f3e4be
Format agent data
ee215c4
Drop agents with only one detection for waymo
428bf1b
More waymo map implementation
33eeb5a
Merge remote-tracking branch 'upstream/main' into main
a8ca28a 
# Conflicts:
#	src/trajdata/dataset_specific/scene_records.py
Refactor and add parallelization to waymo data loading
d786dbd
Add name check for waymo dataset
8537663
Refactor cache_map
01fe8c0
Implement traffic light status extraction from waymo
3371cf0
Revert accidental change
3d2568b
Add tqdm to extract_vectorized function
baffe50
Iterate to get the max/min point in waymo map
3bb3b6a
Set origin shifting to be 0
1653af7
Remove shallow copy
9536359
Update
9c4817e
Remove commented code and add linear interpolation
5cc151b
Refactor waymo data loading
d2a8f7b
Refactor waymo_dataset implementation
6699ad2
Fix major bugs
3aab381
More debugging
fe7470c
Add interpolation and remove hard-coded value
77afe16
Fix interpolation and first/last timesteps
b75bd55
Fix interpolation
abdc20c
Update agent data extraction
f4e72be
Fix first/last_timestep
6256373
remove unnecessary lines
5be1451
@BorisIvanovic
Lots of quality of life improvements in preparation for merging (form…
1431a79 
…atting, a bit of refactoring and speedups, parallelism for maps, etc). Remaining big todo is fixing map element boundaries...
BorisIvanovic and others added 15 commits April 20, 2023 01:41
@BorisIvanovic
Merge branch 'main' of https://github.com/Zhengwai/trajdata into main
dbe21f3
@BorisIvanovic
Merge branch 'main' into main
6f6f279
@BorisIvanovic
Merge pull request #1 from NVlabs/main
a1b6e64 
Syncing with upstream trajdata
@BorisIvanovic
Small formatting post-merge.
9c96d64
@BorisIvanovic
Initial solution for Waymo boundaries.
213c70d
@BorisIvanovic
Saving work quickly to try something else.
4d9ed27
@BorisIvanovic
We'll take it. Just remaining is the connectivity fix and then we can…
83bcdef 
… call it.
@BorisIvanovic
Traffic light saving done.
6f88c1c
@BorisIvanovic
Small comment fix.
efae173
@BorisIvanovic
Fixing lane connectivity, will evaluate it later with improved visual…
2df19b9 
…izations.
@BorisIvanovic
Removing TODO comment.
4e8a307
@BorisIvanovic
Merge branch 'NVlabs:main' into main
296b544
@BorisIvanovic
Lessened the number of hops in the map_api_example.
fddfa26
@BorisIvanovic
Updating version, README, and DATASETS with Waymo info.
09f84d1
@BorisIvanovic
Adding newline.
a0e43b6
@BorisIvanovic BorisIvanovic self-assigned this Apr 22, 2023
BorisIvanovic
BorisIvanovic approved these changes Apr 22, 2023
View reviewed changes
Copy link
Contributor
@BorisIvanovic BorisIvanovic left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Supporting the Waymo Open Dataset has been a long time coming, thank you very much @Zhengwai for working on this!!

@BorisIvanovic BorisIvanovic merged commit 860908a into NVlabs:main Apr 22, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@BorisIvanovic BorisIvanovic BorisIvanovic approved these changes

Assignees

@BorisIvanovic BorisIvanovic

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@songguanyu @BorisIvanovic
0