RangiLyu / nanodet

NanoDet-Plusāš”Super fast and lightweight anchor-free object detection model. šŸ”„Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphonešŸ”„
Apache License 2.0
5.63k stars 1.03k forks source link

Fix class agnostic NMS when #detections exceeds `split_thr` #544

Open TvdrBurgt opened 7 months ago

TvdrBurgt commented 7 months ago

When the number of detections before NMS exceeds the value set by split_thr, NMS would be performed class specifically even if class_agnostic=True. This fix ensures that NMS is always performed class agnostically when class_agnostic=True.