updates
This commit is contained in:
parent
0d770e14df
commit
df3211ba4c
2
test.py
2
test.py
|
@ -125,7 +125,7 @@ def test(
|
|||
iou, bi = bbox_iou(pbox, tbox).max(0)
|
||||
|
||||
# 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
|
||||
detected.append(bi)
|
||||
|
||||
|
|
Loading…
Reference in New Issue