Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates 

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2019-09-01 16:28:25 +02:00
parent c8c0660e6a
commit 9251dfd6a5
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
utils/utils.py
View File

@ -301,8 +301,7 @@ class FocalLoss(nn.Module):
def forward(self, input, target): def forward(self, input, target):
loss = self.loss_fcn(input, target) loss = self.loss_fcn(input, target)
pt = torch.exp(-loss) loss *= self.alpha * (1.000001 - torch.exp(-loss)) ** self.gamma # non-zero power for gradient stability
loss *= self.alpha * (1 - pt) ** self.gamma
if self.reduction == 'mean': if self.reduction == 'mean':
return loss.mean() return loss.mean()