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 typo (#1667)

Browse Source
This commit is contained in:
Xiaolin Wang 2022-06-14 11:20:31 +08:00 committed by GitHub
parent 43b4efb122
commit df434bed69
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mmseg/models/losses/dice_loss.py
View File

@ -35,7 +35,7 @@ def dice_loss(pred,
@weighted_loss @weighted_loss
def binary_dice_loss(pred, target, valid_mask, smooth=1, exponent=2, **kwards): def binary_dice_loss(pred, target, valid_mask, smooth=1, exponent=2, **kwargs):
assert pred.shape[0] == target.shape[0] assert pred.shape[0] == target.shape[0]
pred = pred.reshape(pred.shape[0], -1) pred = pred.reshape(pred.shape[0], -1)
target = target.reshape(target.shape[0], -1) target = target.reshape(target.shape[0], -1)
@ -79,7 +79,7 @@ class DiceLoss(nn.Module):
loss_weight=1.0, loss_weight=1.0,
ignore_index=255, ignore_index=255,
loss_name='loss_dice', loss_name='loss_dice',
**kwards): **kwargs):
super(DiceLoss, self).__init__() super(DiceLoss, self).__init__()
self.smooth = smooth self.smooth = smooth
self.exponent = exponent self.exponent = exponent
@ -94,7 +94,7 @@ class DiceLoss(nn.Module):
target, target,
avg_factor=None, avg_factor=None,
reduction_override=None, reduction_override=None,
**kwards): **kwargs):
assert reduction_override in (None, 'none', 'mean', 'sum') assert reduction_override in (None, 'none', 'mean', 'sum')
reduction = ( reduction = (
reduction_override if reduction_override else self.reduction) reduction_override if reduction_override else self.reduction)