-
Notifications
You must be signed in to change notification settings - Fork 411
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
Torch-based mAP #632
Torch-based mAP #632
Conversation
Very cool! Thanks for porting that. |
Is that correct, that there are two scores while only one prediction @tkupek? |
obviously a bug, you can remove the second score |
Co-authored-by: Tobias Kupek <[email protected]>
Co-authored-by: Tobias Kupek <[email protected]>
@twsl over all it looks Great! mind checking the last failing test for new PT versions: |
* First draft * Remove double score * Calculate num_class only * Support empty predictions * Remove pycocotools from tests * Fix annotation id evals to false if zero * Improve method descriptions * Fix returning metrics if parameters are changed * Apply suggestions from code review Co-authored-by: Tobias Kupek <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Nicki Skafte Detlefsen <[email protected]> Co-authored-by: Justus Schock <[email protected]> Co-authored-by: Jirka Borovec <[email protected]> Co-authored-by: Jirka <[email protected]> Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com> (cherry picked from commit 2494e68)
What does this PR do?
Fixes #587
Simply converted the pycocotools source code to pytorch and torchvision instead of numpy.
The rest in based on @tkupek work.
I plan to rename the file as soon as I know that it's working, right now I still need both for comparison.
Before submitting
PR review
Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in Github issues there's a high chance it will not be merged.
Did you have fun?
Make sure you had fun coding 🙃