Simplify autoshape() post-process (#1603)

* Simplify autoshape() post-process * cleanup
2020-12-09 07:43:58 -08:00
parent 6b1fe3e9dd
commit 61fb2dbd20
4 changed files with 7 additions and 8 deletions
@@ -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: