add yolov3-spp-ultralytics.pt

2020-02-16 23:12:07 -08:00
parent 57798278ad
commit e840b7c781
6 changed files with 27 additions and 26 deletions
@@ -369,7 +369,7 @@ def compute_loss(p, targets, model, giou_flag=True):  # predictions, targets, mo
    tcls, tbox, indices, anchor_vec = build_targets(model, targets)
    h = model.hyp  # hyperparameters
    arc = model.arc  # # (default, uCE, uBCE) detection architectures
-    red = 'sum'  # Loss reduction (sum or mean)
+    red = 'mean'  # Loss reduction (sum or mean)

    # Define criteria
    BCEcls = nn.BCEWithLogitsLoss(pos_weight=ft([h['cls_pw']]), reduction=red)