Skip to content

大佬,发现一个非极大值抑制的问题 #45

@tc191752-oss

Description

@tc191752-oss
def non_maximum_suppression(pred_points):
    """Perform non-maxmum suppression on marking points."""
    suppressed = [False] * len(pred_points)
    for i in range(len(pred_points) - 1):
        for j in range(i + 1, len(pred_points)):
            i_x = pred_points[i][1].x
            i_y = pred_points[i][1].y
            j_x = pred_points[j][1].x
            j_y = pred_points[j][1].y
            # 0.0625 = 1 / 16
            if abs(j_x - i_x) < 0.0625 and abs(j_y - i_y) < 0.0625:

代码这部分定义了一个1/16参数,我理解原图512x512,最终的feature map 是16x16,下采样了32倍,我理解这个1/16应该改为1/32,即在原始图上1个pixel以内的仅保留confidence高的预测点。但是代码中是1/16,是不是搞错了?还是您的意思是想把这个范围设得宽一点:即原图距离2pixel以内的仅保留confidence最高的预测点。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions