-
Notifications
You must be signed in to change notification settings - Fork 50
Open
Description
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
Labels
No labels