updates

2019-03-05 17:10:34 +01:00
parent 2c2d7bc63b
commit a2ad00d6fc
2 changed files with 5 additions and 4 deletions
@@ -379,7 +379,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]