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

delete unnecessary lines of STDCHead (#1231)

Browse Source
This commit is contained in:
MengzhangLI 2022-01-28 10:53:11 +08:00 committed by GitHub
parent 622f28eea3
commit 02d2790145
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/_base_/models/stdc.py
View File

@ -68,7 +68,7 @@ model = dict(
in_index=0, in_index=0,
norm_cfg=norm_cfg, norm_cfg=norm_cfg,
concat_input=False, concat_input=False,
align_corners=False, align_corners=True,
loss_decode=[ loss_decode=[
dict( dict(
type='CrossEntropyLoss', type='CrossEntropyLoss',

5
mmseg/models/decode_heads/stdc_head.py
View File

@ -80,11 +80,6 @@ class STDCHead(FCNHead):
boudary_targets_pyramid[ boudary_targets_pyramid[
boudary_targets_pyramid <= self.boundary_threshold] = 0 boudary_targets_pyramid <= self.boundary_threshold] = 0
seg_logit = F.interpolate(
seg_logit,
boundary_targets.shape[2:],
mode='bilinear',
align_corners=True)
loss = super(STDCHead, self).losses(seg_logit, loss = super(STDCHead, self).losses(seg_logit,
boudary_targets_pyramid.long()) boudary_targets_pyramid.long())
return loss return loss