Skip to content
Snippets Groups Projects
Commit 46898b8e authored by MengzhangLI's avatar MengzhangLI Committed by GitHub
Browse files

delete unnecessary lines of STDCHead (#1231)

parent 83100bb0
No related branches found
No related tags found
No related merge requests found
......@@ -68,7 +68,7 @@ model = dict(
in_index=0,
norm_cfg=norm_cfg,
concat_input=False,
align_corners=False,
align_corners=True,
loss_decode=[
dict(
type='CrossEntropyLoss',
......
......@@ -80,11 +80,6 @@ class STDCHead(FCNHead):
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% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment