Add detection eval optimizations.

argoverse / argoverse-api

Official GitHub repository for Argoverse dataset

https://www.argoverse.org

Other

839 stars 240 forks source link

Add detection eval optimizations. #236

Closed benjaminrwilson closed 3 years ago

benjaminrwilson commented 3 years ago

Speeds up the multiprocessing in the detection eval + transitions classes to dataclasses

johnwlambert commented 3 years ago

Thanks for the PR, @benjaminrwilson. It looks like CI is failing:

FAILED tests/test_eval_detection.py::test_accumulate - TypeError: accumulate(...
FAILED tests/test_eval_detection.py::test_remove_duplicate_instances_ground_truth
FAILED tests/test_eval_detection.py::test_AP_on_filtered_instances - ValueErr...

Do you mind taking a look?