NMS python to c++

def NMS(dects, threshold):
    
    # dects:[x1,y1,x2,y2,score]

    x1 = dects[:,0]
    y1 = dects[:,1]
    x2 = dects[:,2]
    y2 = dects[:,3]
    score = dects[:,4]

    nboxs = dects.shape[0]
    area = (x2-x1+1)*(y2-y1+1)
    order = score.argsort()[::-1] #score从大到小排列的indexs,一维数组

    result = []

    suppressed = np.array([0]*nboxs)

    for _i range(nboxs):
        i = order[_i]
        if suppressed[i] == 1:
            continue
        result.append(i)

        for _j in range(_i+1, nboxs):
            j = order[_j]
            if supressed[j] == 1:
                continue
            xx1 = np.max(x1[i], x1[j])
            yy1 = np.max(y1[i], y1[j])
            xx2 = np.min(x2[i], x2[j])
            yy2 = np.min(y2[i], y2[j])
            w = np.max(0, xx2-xx1+1)
            h = np.max(0, yy2-yy1+1)
            interface = w*h
            overlap = interface / (area[i]+area[j])

            if overlap >= threshold:
                supressed[j] = 1

    return result
                
 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值