Update PanopticQuality.py

PanopticQuality.py

@@ -109,6 +109,7 @@ class PQMetric(evaluate.Metric):
         DEFAULT_STUFF = ["WATER", "SKY", "LAND", "CONSTRUCTION", "ICE", "OWN_BOAT"]
 
         self.label2id = label2id if label2id is not None else DEFAULT_LABEL2ID
+        self.id2label = {id: label for label, id in self.label2id.items()}
         self.stuff =  stuff if stuff is not None else DEFAULT_STUFF
         self.per_class = per_class
         self.split_sq_rq = split_sq_rq
@@ -171,7 +172,6 @@ class PQMetric(evaluate.Metric):
         fn = self.pq_metric.metric.false_negatives.clone()
         iou = self.pq_metric.metric.iou_sum.clone()
 
-        id2label = {id: label for label, id in self.label2id.items()}
         things_stuffs = sorted(self.pq_metric.things) + sorted(self.pq_metric.stuffs)
 
         # compute scores
@@ -182,10 +182,10 @@ class PQMetric(evaluate.Metric):
         if self.per_class:
             if not self.split_sq_rq:
                 result = result.T
-            result_dict["scores"] = {id2label[numeric_label]: result[i].tolist() \
+            result_dict["scores"] = {self.id2label[numeric_label]: result[i].tolist() \
                                      for i, numeric_label in enumerate(things_stuffs)}
             result_dict["scores"].update({"ALL": result.mean(axis=0).tolist()})
-            result_dict["numbers"] =  {id2label[numeric_label]: [tp[i].item(), fp[i].item(), fn[i].item(), iou[i].item()] \
+            result_dict["numbers"] =  {self.id2label[numeric_label]: [tp[i].item(), fp[i].item(), fn[i].item(), iou[i].item()] \
                                        for i, numeric_label in enumerate(things_stuffs)}
             result_dict["numbers"].update({"ALL": [tp.sum().item(), fp.sum().item(), fn.sum().item(), iou.sum().item()]})
         else: