Skip to content

Commit

Permalink
delete unnecessary lines of STDCHead
Browse files Browse the repository at this point in the history
  • Loading branch information
MengzhangLI committed Jan 24, 2022
1 parent bdac9ac commit 8654fe8
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 6 deletions.
2 changes: 1 addition & 1 deletion configs/_base_/models/stdc.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@
in_index=0,
norm_cfg=norm_cfg,
concat_input=False,
align_corners=False,
align_corners=True,
loss_decode=[
dict(
type='CrossEntropyLoss',
Expand Down
5 changes: 0 additions & 5 deletions mmseg/models/decode_heads/stdc_head.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,11 +80,6 @@ def losses(self, seg_logit, seg_label):
boudary_targets_pyramid[
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,
boudary_targets_pyramid.long())
return loss

0 comments on commit 8654fe8

Please sign in to comment.