updates

2018-11-22 14:29:50 +01:00
parent db515a4535
commit f18f288990
2 changed files with 9 additions and 8 deletions
@@ -214,7 +214,8 @@ def build_targets(pred_boxes, pred_conf, pred_cls, target, anchor_wh, nA, nC, nG
        if nTb == 0:
            continue
        t = target[b]
-        FN[b, :nTb] = 1
+        if requestPrecision:
+            FN[b, :nTb] = 1

        # Convert to position relative to box
        TC[b, :nTb], gx, gy, gw, gh = t[:, 0].long(), t[:, 1] * nG, t[:, 2] * nG, t[:, 3] * nG, t[:, 4] * nG