[Enhancement]: fix jpg to png bug when using seg maps (#9078)

* fix jpg->png

* fix jpg->png again

* add suffix

* fix lint

* rm img_suffix

* rename

Co-authored-by: Neesky <[email protected]>

Author: wanghonglie

mmdet/datasets/coco.py

@@ -190,7 +190,7 @@ def _parse_ann_info(self, img_info, ann_info):
         else:
             gt_bboxes_ignore = np.zeros((0, 4), dtype=np.float32)
 
-        seg_map = img_info['filename'].replace('jpg', 'png')
+        seg_map = img_info['filename'].rsplit('.', 1)[0] + self.seg_suffix
 
         ann = dict(
             bboxes=gt_bboxes,

mmdet/datasets/custom.py

@@ -63,6 +63,7 @@ def __init__(self,
                  data_root=None,
                  img_prefix='',
                  seg_prefix=None,
+                 seg_suffix='.png',
                  proposal_file=None,
                  test_mode=False,
                  filter_empty_gt=True,
@@ -71,6 +72,7 @@ def __init__(self,
         self.data_root = data_root
         self.img_prefix = img_prefix
         self.seg_prefix = seg_prefix
+        self.seg_suffix = seg_suffix
         self.proposal_file = proposal_file
         self.test_mode = test_mode
         self.filter_empty_gt = filter_empty_gt