From 3ac51adaef1b0e3005eda4c1c12f3f452d044068 Mon Sep 17 00:00:00 2001 From: zhenming lin <3575188313@qq.com> Date: Thu, 31 Aug 2023 15:55:01 +0800 Subject: [PATCH] fix bugs --- paddleseg/datasets/coco_instance.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/paddleseg/datasets/coco_instance.py b/paddleseg/datasets/coco_instance.py index 76cf00ae89..9be30ce931 100644 --- a/paddleseg/datasets/coco_instance.py +++ b/paddleseg/datasets/coco_instance.py @@ -138,10 +138,6 @@ def __getitem__(self, idx): data['label'] = self._gen_label(gt_polygon, img_h, img_w) data['gt_fields'].append('label') data = self.transforms(data) - if not self.allow_overlap: - data['label'] = data['label'][None] - else: - data['label'] = data['label'].transpose([2, 0, 1]) if self.edge: edge_mask = self._mask2edge(data['label'], radius=2) @@ -231,7 +227,7 @@ def _gen_label(self, gt_polygon, img_h, img_w): label = np.where(mask, cls_idx, label) if self.add_background: - label[label == self.ignore_index] = self.num_classes + label[label == self.ignore_index] = self.num_classes - 1 # multi-label else: label = np.zeros([img_h, img_w, self.num_classes], dtype=np.uint8)