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

feat(metrics): ✨ initial mean average recall metrics added #1656

Closed
wants to merge 8 commits into from
1 change: 0 additions & 1 deletion docs/metrics/mean_average_precision.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
---
comments: true
status: new
---

# Mean Average Precision
Expand Down
18 changes: 18 additions & 0 deletions docs/metrics/mean_average_recall.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
---
comments: true
status: new
---

# Mean Average Recall

<div class="md-typeset">
<h2><a href="#supervision.metrics.mean_average_recall.MeanAverageRecall">MeanAverageRecall</a></h2>
</div>

:::supervision.metrics.mean_average_recall.MeanAverageRecall

<div class="md-typeset">
<h2><a href="#supervision.metrics.mean_average_recall.MeanAverageRecallResult">MeanAverageRecallResult</a></h2>
</div>

:::supervision.metrics.mean_average_recall.MeanAverageRecallResult
3 changes: 2 additions & 1 deletion mkdocs.yml
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,8 @@ nav:
- Core: datasets/core.md
- Utils: datasets/utils.md
- Metrics:
- mAP: metrics/mean_average_precision.md
- Mean Avg. Precision (mAP): metrics/mean_average_precision.md
- Mean Avg. Recall (mAR): metrics/mean_average_recall.md
- Precision: metrics/precision.md
- Recall: metrics/recall.md
- F1 Score: metrics/f1_score.md
Expand Down
4 changes: 4 additions & 0 deletions supervision/metrics/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,10 @@
MeanAveragePrecision,
MeanAveragePrecisionResult,
)
from supervision.metrics.mean_average_recall import (
MeanAverageRecall,
MeanAverageRecallResult,
)
from supervision.metrics.precision import Precision, PrecisionResult
from supervision.metrics.recall import Recall, RecallResult
from supervision.metrics.utils.object_size import (
Expand Down
2 changes: 1 addition & 1 deletion supervision/metrics/mean_average_precision.py
Original file line number Diff line number Diff line change
Expand Up @@ -406,7 +406,7 @@ class MeanAveragePrecisionResult:
Attributes:
metric_target (MetricTarget): the type of data used for the metric -
boxes, masks or oriented bounding boxes.
class_agnostic (bool): When computing class-agnostic results, class ID
is_class_agnostic (bool): When computing class-agnostic results, class ID
is set to `-1`.
mAP_map50_95 (float): the mAP score at IoU thresholds from `0.5` to `0.95`.
mAP_map50 (float): the mAP score at IoU threshold of `0.5`.
Expand Down
Loading