e421e832883241bd7831bf77dc31d5fb31d7da58,mmdet/ops/nms/nms_wrapper.py,,nms,#Any#Any#Any#,9

Before Change

                type(dets)))

    if dets_np.shape[0] == 0:
        inds = []
    else:
        inds = (gpu_nms(dets_np, iou_thr, device_id=device_id)
                if device_id is not None else cpu_nms(dets_np, iou_thr))

    if is_tensor:
        inds = dets.new_tensor(inds, dtype=torch.long)
    else:
        inds = np.array(inds, dtype=np.int64)
    return dets[inds, :], inds


def soft_nms(dets, iou_thr, method="linear", sigma=0.5, min_score=1e-3):

After Change

            inds = nms_cpu.nms(dets_th, iou_thr)

    if is_numpy:
        inds = inds.cpu().numpy()
    return dets[inds, :], inds


def soft_nms(dets, iou_thr, method="linear", sigma=0.5, min_score=1e-3):

Instances