update label2id

PanopticQuality.py

@@ -83,6 +83,7 @@ class PQMetric(evaluate.Metric):
         **kwargs
     ):
         super().__init__(**kwargs)
+        
         DEFAULT_LABEL2ID = {'WATER': 0,
                             'SKY': 1,
                             'LAND': 2,
@@ -96,8 +97,15 @@ class PQMetric(evaluate.Metric):
                             'FLOTSAM': 10,
                             'SAILING_BOAT_WITH_OPEN_SAILS': 11,
                             'CONTAINER': 12,
-                            'PILLAR_BUOY': 13}
+                            'PILLAR_BUOY': 13,
+                            'AERIAL_ANIMAL': 14,
+                            'HUMAN_IN_WATER': 15,
+                            'OWN_BOAT': 16,
+                            'WOODEN_LOG': 17,
+                            'MARITIME_ANIMAL': 18}
+        
         DEFAULT_STUFF = ["WATER", "SKY", "LAND", "CONSTRUCTION", "ICE", "OWN_BOAT"]
+        
         self.label2id = label2id if label2id is not None else DEFAULT_LABEL2ID
         self.stuff =  stuff if stuff is not None else DEFAULT_STUFF
         self.pq_metric = PanopticQuality(