updates
This commit is contained in:
parent
fad618b821
commit
8b707c43c8
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