nGT to nT

2018-09-19 04:32:16 +02:00
parent 29fbcb059f
commit 1cfde4aba8
3 changed files with 7 additions and 7 deletions
@@ -190,7 +190,7 @@ def bbox_iou(box1, box2, x1y1x2y2=True):

 def build_targets(pred_boxes, pred_conf, pred_cls, target, anchor_wh, nA, nC, nG, requestPrecision):
    """
-    returns nGT, nCorrect, tx, ty, tw, th, tconf, tcls
+    returns nT, nCorrect, tx, ty, tw, th, tconf, tcls
    """
    nB = len(target)  # target.shape[0]
    nT = [len(x) for x in target]  # torch.argmin(target[:, :, 4], 1)  # targets per image