From 10ae0b3edad495de2add113ec91d89e1910dd89e Mon Sep 17 00:00:00 2001 From: R-Fehler <36566250+R-Fehler@users.noreply.github.com> Date: Thu, 4 Jan 2024 03:25:35 +0100 Subject: [PATCH] replace '.jpg' instead of 'jpg' to guarantee replacing file ending (#11340) --- mmdet/datasets/api_wrappers/coco_api.py | 2 +- mmdet/datasets/coco_panoptic.py | 2 +- mmdet/evaluation/metrics/coco_panoptic_metric.py | 6 +++--- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/mmdet/datasets/api_wrappers/coco_api.py b/mmdet/datasets/api_wrappers/coco_api.py index 40f7f2c9b93..b2d11a122e1 100644 --- a/mmdet/datasets/api_wrappers/coco_api.py +++ b/mmdet/datasets/api_wrappers/coco_api.py @@ -92,7 +92,7 @@ def createIndex(self) -> None: if 'images' in self.dataset: for img_info in self.dataset['images']: img_info['segm_file'] = img_info['file_name'].replace( - 'jpg', 'png') + '.jpg', '.png') imgs[img_info['id']] = img_info if 'categories' in self.dataset: diff --git a/mmdet/datasets/coco_panoptic.py b/mmdet/datasets/coco_panoptic.py index d5ca7855509..b7a200e01d3 100644 --- a/mmdet/datasets/coco_panoptic.py +++ b/mmdet/datasets/coco_panoptic.py @@ -208,7 +208,7 @@ def parse_data_info(self, raw_data_info: dict) -> dict: if self.data_prefix.get('seg', None): seg_map_path = osp.join( self.data_prefix['seg'], - img_info['file_name'].replace('jpg', 'png')) + img_info['file_name'].replace('.jpg', '.png')) else: seg_map_path = None data_info['img_path'] = img_path diff --git a/mmdet/evaluation/metrics/coco_panoptic_metric.py b/mmdet/evaluation/metrics/coco_panoptic_metric.py index 1554c0908d1..f86be916f9c 100644 --- a/mmdet/evaluation/metrics/coco_panoptic_metric.py +++ b/mmdet/evaluation/metrics/coco_panoptic_metric.py @@ -190,7 +190,7 @@ def gt_to_coco_json(self, gt_dicts: Sequence[dict], } segments_info.append(new_segment_info) - segm_file = image_info['file_name'].replace('jpg', 'png') + segm_file = image_info['file_name'].replace('.jpg', '.png') annotation = dict( image_id=img_id, segments_info=segments_info, @@ -330,7 +330,7 @@ def _compute_batch_pq_stats(self, data_samples: Sequence[dict]): # parse pred img_id = data_sample['img_id'] segm_file = osp.basename(data_sample['img_path']).replace( - 'jpg', 'png') + '.jpg', '.png') result = self._parse_predictions( pred=data_sample, img_id=img_id, @@ -397,7 +397,7 @@ def _process_gt_and_predictions(self, data_samples: Sequence[dict]): # parse pred img_id = data_sample['img_id'] segm_file = osp.basename(data_sample['img_path']).replace( - 'jpg', 'png') + '.jpg', '.png') result = self._parse_predictions( pred=data_sample, img_id=img_id, segm_file=segm_file)