This commit is contained in:
Glenn Jocher 2019-04-18 23:17:51 +02:00
parent fad618b821
commit 8b707c43c8
1 changed files with 1 additions and 1 deletions

View File

@ -125,7 +125,7 @@ def test(
iou, bi = bbox_iou(pbox, tbox).max(0) iou, bi = bbox_iou(pbox, tbox).max(0)
# If iou > threshold and class is correct mark as correct # If iou > threshold and class is correct mark as correct
if iou > iou_thres and bi not in detected and pcls == tcls[bi]: if iou > iou_thres and bi not in detected: # and pcls == tcls[bi]:
correct[i] = 1 correct[i] = 1
detected.append(bi) detected.append(bi)