Skip to content
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.

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

Argoverse2 Dataset Implementation #46

Draft
wants to merge 13 commits into
base: master
Choose a base branch
from

Conversation

aapatni
Copy link

@aapatni aapatni commented Mar 25, 2023

No description provided.

@aapatni aapatni force-pushed the argoverse2_implementation branch from 3fc4a63 to 5e5b331 Compare March 25, 2023 21:49
@bradyz
Copy link
Owner

bradyz commented Mar 27, 2023

hey @aapatni this looks awesome! thanks for working on this.
have you tried training yet? We had tried argoverse quite a while ago as well.
IIRC the diversity and size of dataset was much smaller than nuScenes, so we just settled on that

@aapatni
Copy link
Author

aapatni commented Apr 4, 2023

Hey @bradyz , thanks!

Just got finished on a smaller scale training (150 scenes, 5 epochs) and here are the results. Working on using a larger subset of the scene data + more epochs currently.

pred_sampled

Screenshot 2023-04-04 at 5 28 43 PM

One difference between nuscenes and argoverse2 datasets is that argoverse2 doesn't have the concept of 'visibility' which I believe helps the model understand what objects in the scene are visible to a percentage of the cameras. Currently, we just set the visibility to 100% for all objects in the scene.

@aoyanl
Copy link

aoyanl commented Jun 15, 2024

Hey @bradyz , thanks!

Just got finished on a smaller scale training (150 scenes, 5 epochs) and here are the results. Working on using a larger subset of the scene data + more epochs currently.

pred_sampled pred_sampled

Screenshot 2023-04-04 at 5 28 43 PM

One difference between nuscenes and argoverse2 datasets is that argoverse2 doesn't have the concept of 'visibility' which I believe helps the model understand what objects in the scene are visible to a percentage of the cameras. Currently, we just set the visibility to 100% for all objects in the scene.

Wonderful job! I'm trying the argoverse2 recently. What is the iou of vehicle you get? Looking for your reply!

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.

3 participants