kamata1729/QATM_pytorch

Threshold function always got one match?

Opened this issue · 0 comments

I think below function may be always got at least one matching.

def nms(score, w_ini, h_ini, thresh=0.7):
    dots = np.array(np.where(score > thresh*score.max()))

may be need change "score.max()" to a constant (0.1~0.2).