mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

remove gt padding

pull/2290/head
xiexinch 2022-11-19 17:38:10 +08:00
parent 8fed7f543f
commit f915df4b0c
2 changed files with 1 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mmseg/models/data_preprocessor.py
View File

@ -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,

10
mmseg/models/segmentors/base.py
View File

@ -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