Spaces:
Build error
Build error
Commit
·
7e28a06
1
Parent(s):
f28256c
update docstring
Browse files- PanopticQuality.py +3 -3
PanopticQuality.py
CHANGED
|
@@ -53,8 +53,8 @@ Args:
|
|
| 53 |
The last dimension should hold the category index at position 0, and
|
| 54 |
the instance ID at position 1.
|
| 55 |
Returns:
|
| 56 |
-
A
|
| 57 |
-
|
| 58 |
Examples:
|
| 59 |
>>> import evaluate
|
| 60 |
>>> from seametrics.payload.processor import PayloadProcessor
|
|
@@ -170,7 +170,7 @@ class PQMetric(evaluate.Metric):
|
|
| 170 |
|
| 171 |
def _compute(self, *, predictions, references, **kwargs):
|
| 172 |
"""Called within the evaluate.Metric.compute() method"""
|
| 173 |
-
tp = self.pq_metric.metric.true_positives.clone().cpu()
|
| 174 |
fp = self.pq_metric.metric.false_positives.clone().cpu()
|
| 175 |
fn = self.pq_metric.metric.false_negatives.clone().cpu()
|
| 176 |
iou = self.pq_metric.metric.iou_sum.clone().cpu()
|
|
|
|
| 53 |
The last dimension should hold the category index at position 0, and
|
| 54 |
the instance ID at position 1.
|
| 55 |
Returns:
|
| 56 |
+
A dictionary containing PQ, RQ, SQ scores for key "scores" and
|
| 57 |
+
FP, TP, FN, IOU sum numbers for key "numbers".
|
| 58 |
Examples:
|
| 59 |
>>> import evaluate
|
| 60 |
>>> from seametrics.payload.processor import PayloadProcessor
|
|
|
|
| 170 |
|
| 171 |
def _compute(self, *, predictions, references, **kwargs):
|
| 172 |
"""Called within the evaluate.Metric.compute() method"""
|
| 173 |
+
tp = self.pq_metric.metric.true_positives.clone().cpu() # shape : (area_rngs, n_classes (sorted things + sorted stuffs))
|
| 174 |
fp = self.pq_metric.metric.false_positives.clone().cpu()
|
| 175 |
fn = self.pq_metric.metric.false_negatives.clone().cpu()
|
| 176 |
iou = self.pq_metric.metric.iou_sum.clone().cpu()
|