Update PanopticQuality.py

PanopticQuality.py

@@ -113,6 +113,7 @@ class PQMetric(evaluate.Metric):
         self.stuff =  stuff if stuff is not None else DEFAULT_STUFF
         self.per_class = per_class
         self.split_sq_rq = split_sq_rq
+        self.things_stuffs = sorted(self.pq_metric.things) + sorted(self.pq_metric.stuffs)
         self.pq_metric = PanopticQuality(
             things=set([self.label2id[label] for label in self.label2id.keys() if label not in self.stuff]), 
             stuffs=set([self.label2id[label] for label in self.label2id.keys() if label in self.stuff]),
@@ -172,8 +173,6 @@ class PQMetric(evaluate.Metric):
         fn = self.pq_metric.metric.false_negatives.clone()
         iou = self.pq_metric.metric.iou_sum.clone()
 
-        things_stuffs = sorted(self.pq_metric.things) + sorted(self.pq_metric.stuffs)
-
         # compute scores
         result = self.pq_metric.compute() # shape : (n_classes (sorted things + sorted stuffs), scores (pq, sq, rq))
 
@@ -183,10 +182,10 @@ class PQMetric(evaluate.Metric):
             if not self.split_sq_rq:
                 result = result.T
             result_dict["scores"] = {self.id2label[numeric_label]: result[i].tolist() \
-                                     for i, numeric_label in enumerate(things_stuffs)}
+                                     for i, numeric_label in enumerate(self.things_stuffs)}
             result_dict["scores"].update({"ALL": result.mean(axis=0).tolist()})
             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)}
+                                       for i, numeric_label in enumerate(self.things_stuffs)}
             result_dict["numbers"].update({"ALL": [tp.sum().item(), fp.sum().item(), fn.sum().item(), iou.sum().item()]})
         else:
             result_dict["scores"] = {"ALL": result.tolist() if self.split_sq_rq else [result.tolist()]}