e4d62de5bc12d1e411adbfe4b76f15d157d77c65,utils/utils.py,,non_max_suppression,#Any#Any#Any#,295

Before Change

                    ious = bbox_iou(det_class[:1], det_class[1:])

                    if ious.max() > 0.5:
                        det_max.append(det_class[0].unsqueeze(0))

                    // Remove detections with IoU >= NMS threshold
                    det_class = det_class[1:][ious < nms_thres]

After Change

                while len(dc) > 1:
                    iou = bbox_iou(dc[:1], dc[1:])  // iou with other boxes
                    if iou.max() > 0.5:
                        det_max.append(dc[:1])
                    dc = dc[1:][iou < nms_thres]  // remove ious > threshold

            elif nms_style == "MERGE":  // weighted mixture box
                while len(dc) > 0:

Instances