search

facebookresearch / detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
https://detectron2.readthedocs.io/en/latest/
Apache License 2.0
30.24k stars 7.44k forks source link

The latest version replaces nms with soft-nms #3762

Open ts19970816 opened 2 years ago

ts19970816 commented 2 years ago

The latest version replaces nms with soft-nms

shenyi0220 commented 2 years ago

Here's a cluster better than prev NMS-versions(https://github.com/shenyi0220/CP-Cluster)

shubhamgajrani commented 2 years ago

(https://github.com/facebookresearch/detectron2/tree/main/detectron2/layers)/nms.py -- still uses NMS instead of soft NMS. How can I use soft-NMS for predictions?