Slow NMS in Python
macwilam opened this issue · 0 comments
macwilam commented
Hi in two python implementations you use your own NMS function (in centerface.py fiels) . This implementation seems to be fairly slow.
I think the implementation from faster rcnn mentioned here:
pytorch/vision#942 (comment)
is basically a drop in replacement for you implementation and is much faster. The speed difference is stronger the more boxes you find (and sometimes you algorithm fins a lot of them).
Kind regards