From add835be3fe3648962d2518a355badcdecea56d1 Mon Sep 17 00:00:00 2001 From: panfeng-hover <48805111+panfeng-hover@users.noreply.github.com> Date: Tue, 12 Apr 2022 03:26:25 -0700 Subject: [PATCH] Avoid converting label ids twice by label map during evaluation (#1417) * Avoid converting label ids twice by label map during evaluation * Update mmseg/datasets/custom.py Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com> * lint Co-authored-by: Miao Zheng <76149310+MeowZheng@users.noreply.github.com> Co-authored-by: MeowZheng <meowzheng@outlook.com> --- mmseg/datasets/custom.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/mmseg/datasets/custom.py b/mmseg/datasets/custom.py index 2649a581..4615d411 100644 --- a/mmseg/datasets/custom.py +++ b/mmseg/datasets/custom.py @@ -298,9 +298,18 @@ class CustomDataset(Dataset): for pred, index in zip(preds, indices): seg_map = self.get_gt_seg_map_by_idx(index) pre_eval_results.append( - intersect_and_union(pred, seg_map, len(self.CLASSES), - self.ignore_index, self.label_map, - self.reduce_zero_label)) + intersect_and_union( + pred, + seg_map, + len(self.CLASSES), + self.ignore_index, + # as the labels has been converted when dataset initialized + # in `get_palette_for_custom_classes ` this `label_map` + # should be `dict()`, see + # https://github.com/open-mmlab/mmsegmentation/issues/1415 + # for more ditails + label_map=dict(), + reduce_zero_label=self.reduce_zero_label)) return pre_eval_results @@ -417,7 +426,7 @@ class CustomDataset(Dataset): num_classes, self.ignore_index, metric, - label_map=self.label_map, + label_map=dict(), reduce_zero_label=self.reduce_zero_label) # test a list of pre_eval_results else: -- GitLab