Spaces:
Sleeping
Sleeping
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()
|