search

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.61k stars 1.03k forks source link

Issue with Multiple bounding box on detection.. #405

Open LolikaPadmanbhan opened 2 years ago

LolikaPadmanbhan commented 2 years ago

Hi, when i tried training nanodet-m on my custom data.. but i end up getting multiple detection for each class.. Why is that ,, ? And also is possible to replace NMS in nanodet? Kindly help me with the answer. Thanking in advance.

Ljung323 commented 9 months ago

getting multiple detection for each class

enabling class_agnostic and setting lower nms_thr in nms may solve the problem? https://github.com/RangiLyu/nanodet/blob/226d8cc45b92313b625e893d88c595b7856f556c/nanodet/model/module/nms.py#L90-L92 https://github.com/RangiLyu/nanodet/blob/226d8cc45b92313b625e893d88c595b7856f556c/nanodet/model/module/nms.py#L16