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

Fixed a few spelling errors (#1656)

Browse Source 
* 修复拼写错误

* Fix lint error

* Fix lint error

* Fix lint error
This commit is contained in:
rotorliu 2022-06-14 12:10:59 +08:00 committed by GitHub
parent df434bed69
commit f3c72c6dbb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -67,19 +67,19 @@ class STDCHead(FCNHead):
boundary_targets_x4_up[
boundary_targets_x4_up <= self.boundary_threshold] = 0
boudary_targets_pyramids = torch.stack(
boundary_targets_pyramids = torch.stack(
(boundary_targets, boundary_targets_x2_up, boundary_targets_x4_up),
dim=1)
boudary_targets_pyramids = boudary_targets_pyramids.squeeze(2)
boudary_targets_pyramid = F.conv2d(boudary_targets_pyramids,
self.fusion_kernel)
boundary_targets_pyramids = boundary_targets_pyramids.squeeze(2)
boundary_targets_pyramid = F.conv2d(boundary_targets_pyramids,
self.fusion_kernel)
boudary_targets_pyramid[
boudary_targets_pyramid > self.boundary_threshold] = 1
boudary_targets_pyramid[
boudary_targets_pyramid <= self.boundary_threshold] = 0
boundary_targets_pyramid[
boundary_targets_pyramid > self.boundary_threshold] = 1
boundary_targets_pyramid[
boundary_targets_pyramid <= self.boundary_threshold] = 0
loss = super(STDCHead, self).losses(seg_logit,
boudary_targets_pyramid.long())
boundary_targets_pyramid.long())
return loss