From f915df4b0cb2ed431bba60f75f07c5453207e833 Mon Sep 17 00:00:00 2001 From: xiexinch Date: Sat, 19 Nov 2022 17:38:10 +0800 Subject: [PATCH] remove gt padding --- mmseg/models/data_preprocessor.py | 1 - mmseg/models/segmentors/base.py | 10 +--------- 2 files changed, 1 insertion(+), 10 deletions(-) diff --git a/mmseg/models/data_preprocessor.py b/mmseg/models/data_preprocessor.py index 082499983..4b94a00d7 100644 --- a/mmseg/models/data_preprocessor.py +++ b/mmseg/models/data_preprocessor.py @@ -139,7 +139,6 @@ class SegDataPreProcessor(BaseDataPreprocessor): if self.test_cfg: inputs, data_samples = stack_batch( inputs=inputs, - data_samples=data_samples, size=self.test_cfg.get('size', None), size_divisor=self.test_cfg.get('size_divisor', None), pad_val=self.pad_val, diff --git a/mmseg/models/segmentors/base.py b/mmseg/models/segmentors/base.py index 3c99eaf3c..9fd496f27 100644 --- a/mmseg/models/segmentors/base.py +++ b/mmseg/models/segmentors/base.py @@ -165,11 +165,6 @@ class BaseSegmentor(BaseModel, metaclass=ABCMeta): i_seg_logits = seg_logits[i:i + 1, :, padding_top:H - padding_bottom, padding_left:W - padding_right] - i_gt_sem_seg = data_samples[i].gt_sem_seg.data[:, - padding_top:H - - padding_bottom, - padding_left:W - - padding_right] # resize as original shape i_seg_logits = resize( @@ -190,10 +185,7 @@ class BaseSegmentor(BaseModel, metaclass=ABCMeta): 'seg_logits': PixelData(**{'data': i_seg_logits}), 'pred_sem_seg': - PixelData(**{'data': i_seg_pred}), - 'gt_sem_seg': - PixelData() if only_prediction else PixelData( - **{'data': i_gt_sem_seg}) + PixelData(**{'data': i_seg_pred}) }) return data_samples