Simplify autoshape() post-process (#1653)

* Simplify autoshape() post-process

* cleanup

* cleanup
This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-12-09 07:44:06 -08:00
1 parent 84f9bb5d92
commit fa8f1fb0e9
4 files changed
+9 -10

No files matched your search

+1 -1
View File
@@ -258,7 +258,7 @@ def wh_iou(wh1, wh2):
return inter / (wh1.prod(2) + wh2.prod(2) - inter) # iou = inter / (area1 + area2 - inter)
def non_max_suppression(prediction, conf_thres=0.1, iou_thres=0.6, classes=None, agnostic=False, labels=()):
def non_max_suppression(prediction, conf_thres=0.25, iou_thres=0.45, classes=None, agnostic=False, labels=()):
"""Performs Non-Maximum Suppression (NMS) on inference results
Returns: