updates

2019-02-21 10:57:55 +01:00 · 2019-02-21 10:57:55 +01:00 · 646a21a5cd
parent 7a31f4b288
commit 646a21a5cd
1 changed files with 2 additions and 1 deletions
--- a/models.py
+++ b/models.py
@ -159,7 +159,8 @@ class YOLOLayer(nn.Module):
            # Compute losses
            nT = sum([len(x) for x in targets])  # number of targets
            nM = mask.sum().float()  # number of anchors (assigned to targets)
-            k = nM / bs
+            k = 1 #nM / bs
+
            if nM > 0:
                lxy = k * MSELoss(xy[mask], txy[mask])
                lwh = k * MSELoss(wh[mask], twh[mask])