fix bug where own_boat and construction classes are confused in results

PanopticQuality.py

@@ -124,7 +124,7 @@ class PQMetric(evaluate.Metric):
             return_sq_and_rq=split_sq_rq,
             areas=area_rng
         )
-        self.things_stuffs = sorted(self.pq_metric.things) + sorted(self.pq_metric.stuffs)
+        self.cont_to_cat = {label:key for key, label in self.pq_metric.metric.cat_id_to_continuous_id.items()}
 
     def _info(self):
         return evaluate.MetricInfo(
@@ -187,10 +187,10 @@ class PQMetric(evaluate.Metric):
             if not self.split_sq_rq:
                 result = result.unsqueeze(0)
             result_dict["scores"] = {self.id2label[numeric_label]: result[:,:, i].numpy() \
-                                     for i, numeric_label in enumerate(self.things_stuffs)}
+                                     for i, numeric_label in self.cont_to_cat.items()}
             result_dict["scores"].update({"ALL": result.mean(dim=-1).numpy()})
             result_dict["numbers"] =  {self.id2label[numeric_label]: np.stack([tp[:, i].numpy(), fp[:, i].numpy(), fn[:, i].numpy(), iou[:, i].numpy()])\
-                                       for i, numeric_label in enumerate(self.things_stuffs)}
+                                       for i, numeric_label in self.cont_to_cat.items()}
             result_dict["numbers"].update({"ALL": np.stack([tp.sum(dim=1).numpy(), fp.sum(dim=1).numpy(), fn.sum(dim=1).numpy(), iou.sum(dim=1).numpy()])})
         else:
             result_dict["scores"] = {"ALL": result.numpy() if self.split_sq_rq else (result.numpy()[np.newaxis, ...] if len(self.pq_metric.get_areas())>1 else result.numpy()[np.newaxis, np.newaxis, ...])}