ef58bc62d3b5ee992b37e9277a7aa7701d8c1757,mmdet/ops/nms/nms_wrapper.py,,soft_nms,#Any#Any#Any#Any#Any#,64

Before Change

    method_codes = {"linear": 1, "gaussian": 2}
    if method not in method_codes:
        raise ValueError("Invalid method for SoftNMS: {}".format(method))
    new_dets, inds = soft_nms_cpu(
        dets_np,
        iou_thr,
        method=method_codes[method],

After Change

        dets_t = dets.detach().cpu()
    elif isinstance(dets, np.ndarray):
        is_tensor = False
        dets_t = torch.from_numpy(dets)
    else:
        raise TypeError(
            "dets must be either a Tensor or numpy array, but got {}".format(
                type(dets)))

    method_codes = {"linear": 1, "gaussian": 2}
    if method not in method_codes:
        raise ValueError("Invalid method for SoftNMS: {}".format(method))
    results = nms_cpu.soft_nms(dets_t, iou_thr, method_codes[method], sigma,
                               min_score)

    new_dets = results[:, :5]
    inds = results[:, 5]

    if is_tensor:
        return dets.new_tensor(new_dets), dets.new_tensor(
            inds, dtype=torch.long)

Instances