sshaoshuai / MTR

MTR: Motion Transformer with Global Intention Localization and Local Movement Refinement, NeurIPS 2022.
Apache License 2.0
682 stars 108 forks source link

Argoverse 2 #10

Closed Cram3r95 closed 1 year ago

Cram3r95 commented 1 year ago

Is this code prepared to run in Argoverse2?

sshaoshuai commented 1 year ago

For supporting other datasets like argoverse2, you just need to write a specific data loader for them. For now, we haven t supported argoverse2 since Waymo dataset is larger and the performance is more stable.

Cram3r95 commented 1 year ago

Actually Argoverse2 is bigger and better than Waymo:

imagen

Ok, thank you so much @sshaoshuai, I will try to adapt the code.

sshaoshuai commented 1 year ago

It seems that the dataset size of waymo motion prediction (> 1 TB) is much larger than other datasets, and I think it may provide more information than other datasets (e.g., more detailed HDMap). You can try both and see which one you prefer.

Cram3r95 commented 1 year ago

Ok, I will try, quite sure Argoverse2 has a more detailed HD info.

sshaoshuai commented 1 year ago

Feel free to reopen it if there are further questions.

ares89 commented 1 year ago

@Cram3r95 Have you migrated to argoverse2 yet?

Cram3r95 commented 1 year ago

@ares89 not with this algorithm, but it is easy to adapt the graph from Argo1 to Argo2 format