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

F1 metric #1521

Draft
wants to merge 4 commits into
base: develop
Choose a base branch
from
Draft

F1 metric #1521

wants to merge 4 commits into from

Conversation

LinasKo
Copy link
Collaborator

@LinasKo LinasKo commented Sep 18, 2024

Description

This PR adds the F1 score as a metric.

Other changes:

  • Removed Internal Data store (dead code)
  • Small refactoring of mAP metric.
  • One renamed an mAP parameter to ap_per_class
  • One new mAP parameter mAP_scores.

Type of change

Please delete options that are not relevant.

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • This change requires a documentation update

How has this change been tested, please provide a testcase or example of how you tested the change?

I ran it on a small set of tests locally:

  • No detections, one image, multiple sets of detections, multiple averaging types, boxes & masks.

TODO: units tests
TODO: Colab
TODO: check docs

Any specific deployment considerations

Docs

  • Docs updated? What were the changes:

@LinasKo
Copy link
Collaborator Author

LinasKo commented Sep 18, 2024

Hey @onuralpszr, if you have some time, could you give this a glance and see if anything weird stands out?

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.

1 participant