search

ultralytics / yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
https://docs.ultralytics.com
GNU Affero General Public License v3.0
50.25k stars 16.23k forks source link

NMS #813

Closed SISTMrL closed 3 years ago

SISTMrL commented 4 years ago

❔Question

Hello, I read the code of NMS and find that's just NMS, do you have a try to use other nms, such as:soft nms, DIOU nms etc.

Additional context

glenn-jocher commented 4 years ago

@SISTMrL we don't have soft-nms, but if you'd like to submit a PR please do!

abhiagwl4262 commented 4 years ago

hey @glenn-jocher, You had a long discussion here. Did you try any of the soft-NMS or DIOU-ClusterNMS ?

glenn-jocher commented 4 years ago

@abhiagwl4262 a detailed analysis of various NMS techniques with YOLOv5 by the DIoU and ClusterNMS author is here: https://github.com/Zzh-tju/yolov5

github-actions[bot] commented 3 years ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.

shenyi0220 commented 2 years ago

A better option of bounding box fusion is provided here(https://github.com/shenyi0220/CP-Cluster)

Ankit12bhatt commented 10 months ago

@shenyi0220 I want to apply it to yolov5, and I have no idea how to do it.