Python：实现非极大值抑制

def NMS_points(heatmap,threshold,nms_dist):
    """
    input:
        heatmap: size = [H,W]
        threshold: if heatmap[i]>threshold, then it's an interest point
        nms_dist: distance to suppress
    output:
        heatmap_nms: size = [3,N]
    """

    # Step1: 在heatmap中找到大于阈值的兴趣点，组成pts(3,n),分别记录位置和value
    # Step2：根据nms_dist各扩展heatmap各边，组成grid，对应兴趣点为1，否则为0     np.pad函数
    # Step3：从大到小排序pts，并按照这个顺序，在grid中找到对应位置的点，值设为-1；将附近pad内的点值都设为0
    #       即值为1，还没处理；值为0，不是兴趣点；值为-1，是兴趣点
    # Step4：再把真正兴趣点的位置提出来
    pass