search

youngwanLEE / centermask2

[CVPR 2020] CenterMask : Real-time Anchor-Free Instance Segmentation
Other
778 stars 159 forks source link

SCORE_THRESH_TEST = 0.05 doesn't work #78

Open haderalim opened 3 years ago

haderalim commented 3 years ago

In 'centermask/modeling/fcos/fcos_outputs/'

self.pre_nms_thresh = cfg.SCORE_THRESH_TEST

candidate_inds = box_cls > self.pre_nms_thresh pre_nms_top_n = candidate_inds.reshape(N, -1).sum(1)

self.pre_nms_thresh = 0.05 is the default value and this make the value of pre_nms_top_n = tensor([0, 0]. So, it doesn't work well. How can I fix this problem?

haderalim commented 3 years ago

@youngwanLEE