search

hujiecpp / ISTR

ISTR: End-to-End Instance Segmentation with Transformers (https://arxiv.org/abs/2105.00637)
202 stars 28 forks source link

NMS module? #7

Closed cao13jf closed 3 years ago

cao13jf commented 3 years ago

Hi @hujiecpp, thank you for making this project publicly available. I tried to apply this framework to an instance segmentation task of medical images. The prediction result looks good but some objects seem to be detected as multiple overlapped objects. This problem is solved by NMS in MASK RCNN. I checked the inference code but failed to find the way to overcome this problem. Is there any counterpart (like NMS) in ISTR to filter out extra predictions? I probably missed it in the manuscirpt, sorry about that. In advance thank you.

xiaozhen18239228065 commented 2 years ago

@cao13jf Hello, I seem to have encountered the same problem as you. How did you solve it in the end.