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

updates

This commit is contained in:
Glenn Jocher 2018-11-22 14:33:01 +01:00
parent f18f288990
commit bec94be01a
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
models.py
View File

@ -254,8 +254,11 @@ class Darknet(nn.Module):
output.append(x)
layer_outputs.append(x)
if is_training and requestPrecision:
self.losses['nT'] /= 3
self.losses['TP'] = 0
self.losses['FP'] = 0
self.losses['FN'] = 0
if is_training and requestPrecision:
self.losses['TC'] /= 3 # target category
metrics = torch.zeros(3, len(self.losses['FPe'])) # TP, FP, FN