updates

2019-02-27 12:32:25 +01:00
parent 9a27339e04
commit 358f34afa8
2 changed files with 6 additions and 4 deletions
@@ -374,7 +374,7 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.4):
        if prediction.is_cuda:
            unique_labels = unique_labels.cuda(prediction.device)

-        nms_style = 'MERGE'  # 'OR' (default), 'AND', 'MERGE' (experimental)
+        nms_style = 'OR'  # 'OR' (default), 'AND', 'MERGE' (experimental)
        for c in unique_labels:
            # Get the detections with class c
            dc = detections[detections[:, -1] == c]