add per class pq score + sq and rq split up

PanopticQuality.py

@@ -151,6 +151,9 @@ class PQMetric(evaluate.Metric):
         # in case the inputs are lists, convert them to numpy arrays
 
         self.pq_metric.update(prediction, reference)
+        print("TP:", self.pq_metric.metric.true_positives)
+        print("FP:", self.pq_metric.metric.false_positives)
+        print("FN:", self.pq_metric.metric.false_negatives)
 
         # does not impact the metric, but is required for the interface x_x
         super(evaluate.Metric, self).add(
@@ -161,7 +164,11 @@ class PQMetric(evaluate.Metric):
 
     def _compute(self, *, predictions, references, **kwargs):
         """Called within the evaluate.Metric.compute() method"""
-        return self.pq_metric.compute()
+        result = self.pq_metric.compute() # n_classes (sorted things + sorted stuffs), (pq, sq, rq)
+        id2label = {id: label for label, id in self.label2id.items()}
+        return {
+            id2label[numeric_label]: result[i] for i, numeric_label in enumerate(self.pq_metric.things + self.pq_metric.stuffs)
+        }
 
     def add_payload(self, payload: Payload, model_name: str = None):
         """Converts the payload to the format expected by the metric"""