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)
|
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)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue