franzi2505 commited on
Commit
7e28a06
·
1 Parent(s): f28256c

update docstring

Browse files
Files changed (1) hide show
  1. 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 single float number in range [0, 1] that represents the PQ score.
57
- 1 is perfect panoptic segmentation, 0 is worst possible panoptic segmentation.
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()