Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Feature request - clustered NMS (in addition to CIoU) #16

Open
lessw2020 opened this issue May 10, 2020 · 1 comment
Open

Feature request - clustered NMS (in addition to CIoU) #16

lessw2020 opened this issue May 10, 2020 · 1 comment
Labels
enhancement New feature or request

Comments

@lessw2020
Copy link

Just wanted to put this on the radar for the project - the developers of Complete IoU have extended it with clustered NMS:
"...we propose Cluster-NMS, where NMS during inference is done by implicitly clustering detected boxes and usually requires less iterations. Cluster-NMS is very efficient due to its pure GPU implementation, , and geometric factors can be incorporated to improve both AP and AR. In the experiments, CIoU loss and Cluster-NMS have been applied to state-of-the-art instance segmentation (e.g., YOLACT), and object detection (e.g., YOLO v3, SSD and Faster R-CNN) models. Taking YOLACT on MS COCO as an example, our method achieves performance gains as 1.7 AP and 6.2 AR100 for object detection..."

https://arxiv.org/abs/2005.03572
code:
https://github.com/Zzh-tju/CIoU

@rwightman rwightman added the enhancement New feature or request label May 26, 2020
@hamadichihaoui
Copy link

#52

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

3 participants