mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

fix get gt data

Browse Source
This commit is contained in:
xiexinch 2022-11-10 14:28:20 +08:00
parent 70daaaad59
commit d0b35cdacd
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -165,10 +165,11 @@ 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[:, 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(