minor fix

Author: BIGWangYuDong

configs/_base_/datasets/objects365v1_detection.py

@@ -59,5 +59,6 @@
     type='CocoMetric',
     ann_file=data_root + 'annotations/objects365_val.json',
     metric='bbox',
+    sort_categories=True,
     format_only=False)
 test_evaluator = val_evaluator

mmdet/datasets/objects365.py

@@ -106,20 +106,12 @@ def load_data_list(self) -> List[dict]:
             self.coco = self.COCOAPI(local_path)
 
         # 'categories' list in objects365_train.json and objects365_val.json
-        # is inconsistent, need sorted list(or dict) before get cat_ids.
+        # is inconsistent, need sort list(or dict) before get cat_ids.
         cats = self.coco.cats
         sorted_cats = {i: cats[i] for i in sorted(cats)}
-        print('----')
-        print(cats)
-        print('----')
-        print(sorted_cats)
         self.coco.cats = sorted_cats
         categories = self.coco.dataset['categories']
         sorted_categories = sorted(categories, key=lambda i: i['id'])
-        print('----')
-        print(categories)
-        print('----')
-        print(sorted_categories)
         self.coco.dataset['categories'] = sorted_categories
         # The order of returned `cat_ids` will not
         # change with the order of the `classes`

mmdet/evaluation/metrics/coco_metric.py

@@ -59,6 +59,8 @@ class CocoMetric(BaseMetric):
             names to disambiguate homonymous metrics of different evaluators.
             If prefix is not provided in the argument, self.default_prefix
             will be used instead. Defaults to None.
+        sort_categories (bool): Whether sort categories in annotations. Only
+            used for `Objects365V1Dataset`. Defaults to False.
     """
     default_prefix: Optional[str] = 'coco'
 
@@ -73,7 +75,8 @@ def __init__(self,
                  outfile_prefix: Optional[str] = None,
                  file_client_args: dict = dict(backend='disk'),
                  collect_device: str = 'cpu',
-                 prefix: Optional[str] = None) -> None:
+                 prefix: Optional[str] = None,
+                 sort_categories: bool = False) -> None:
         super().__init__(collect_device=collect_device, prefix=prefix)
         # coco evaluation metrics
         self.metrics = metric if isinstance(metric, list) else [metric]
@@ -112,6 +115,17 @@ def __init__(self,
         if ann_file is not None:
             with self.file_client.get_local_path(ann_file) as local_path:
                 self._coco_api = COCO(local_path)
+                if sort_categories:
+                    # 'categories' list in objects365_train.json and
+                    # objects365_val.json is inconsistent, need sort
+                    # list(or dict) before get cat_ids.
+                    cats = self._coco_api.cats
+                    sorted_cats = {i: cats[i] for i in sorted(cats)}
+                    self._coco_api.cats = sorted_cats
+                    categories = self._coco_api.dataset['categories']
+                    sorted_categories = sorted(
+                        categories, key=lambda i: i['id'])
+                    self._coco_api.dataset['categories'] = sorted_categories
         else:
             self._coco_api = None