[Feature] Add panoptic segmentation metric (#2230)

* panoptic segmentation metric * fix * update * update * fix * fix * fix * fix * rename * merge * merge * fix test * update * update * update
open-mmlab · Feb 15, 2023 · 4cfae3f · 4cfae3f
1 parent f4ae6d7
commit 4cfae3f
Show file tree

Hide file tree

Showing 7 changed files with 717 additions and 6 deletions.
diff --git a/mmdet3d/evaluation/__init__.py b/mmdet3d/evaluation/__init__.py
@@ -6,10 +6,11 @@
  eval_det_cls, eval_map_recall, fast_hist, get_acc,
  get_acc_cls, get_classwise_aps, get_single_class_aps,
  indoor_eval, instance_seg_eval, load_lyft_gts,
- load_lyft_predictions, lyft_eval, per_class_iou,
- rename_gt, seg_eval)
+ load_lyft_predictions, lyft_eval, panoptic_seg_eval,
+ per_class_iou, rename_gt, seg_eval)
 from .metrics import (IndoorMetric, InstanceSegMetric, KittiMetric, LyftMetric,
- NuScenesMetric, SegMetric, WaymoMetric)
+ NuScenesMetric, PanopticSegMetric, SegMetric,
+ WaymoMetric)
 
 __all__ = [
  'kitti_eval_coco_style', 'kitti_eval', 'indoor_eval', 'lyft_eval',
@@ -19,5 +20,6 @@
  'get_classwise_aps', 'get_single_class_aps', 'fast_hist', 'per_class_iou',
  'get_acc', 'get_acc_cls', 'seg_eval', 'KittiMetric', 'NuScenesMetric',
  'IndoorMetric', 'LyftMetric', 'SegMetric', 'InstanceSegMetric',
- 'WaymoMetric', 'eval_class', 'do_eval'
+ 'WaymoMetric', 'eval_class', 'do_eval', 'PanopticSegMetric',
+ 'panoptic_seg_eval'
 ]
diff --git a/mmdet3d/evaluation/functional/__init__.py b/mmdet3d/evaluation/functional/__init__.py
@@ -6,6 +6,7 @@
 from .kitti_utils import do_eval, kitti_eval, kitti_eval_coco_style
 from .lyft_eval import (get_classwise_aps, get_single_class_aps, load_lyft_gts,
  load_lyft_predictions, lyft_eval)
+from .panoptic_seg_eval import panoptic_seg_eval
 from .scannet_utils import evaluate_matches, scannet_eval
 from .seg_eval import fast_hist, get_acc, get_acc_cls, per_class_iou, seg_eval
 
@@ -15,5 +16,5 @@
  'load_lyft_predictions', 'lyft_eval', 'get_classwise_aps',
  'get_single_class_aps', 'fast_hist', 'per_class_iou', 'get_acc',
  'get_acc_cls', 'seg_eval', 'kitti_eval', 'kitti_eval_coco_style',
- 'scannet_eval', 'evaluate_matches', 'do_eval'
+ 'scannet_eval', 'evaluate_matches', 'do_eval', 'panoptic_seg_eval'
 ]