Conf_thres and iou_thres

[tdhooghe]

Search before asking

• [X] I have searched the YOLOv5 issues and discussions and found no similar questions.

Question

Hi Glenn, others,

I wondered how you arrived at a .25 conf_thres and .45 iou_thres for the detect.py and model_hub implementations. Did you by any chance do a study that I can look into?

Kind regards,

Additional

No response

[glenn-jocher]

@tdhooghe these are detection standard values across a few different tools, i.e. CoreML uses the same thresholds by default.

[tdhooghe]

Thank you for your quick response! Could you maybe point me into the direction of which other tools use this threshold and why these values are accepted in general?

[creativesalam]

@tdhooghe It depends on your application which values work better for you. Default might work for most people but it is not guaranteed that these values are better for all applications. E.g. I use conf_thres=0.6 in my current project because for my problem False Positive is very dangerous. Higher conf_thres will suppress false positive but at the cost of missing some detections and vice versa. It is up to you to decide what better suits to your business problem.

P.S. False positive can also be suppressed by adding more background images but it is just for your understanding. You can always tweak these values to get your desirable outcome.

[github-actions[bot]]

👋 Hello, this issue has been automatically marked as stale because it has not had recent activity. Please note it will be closed if no further activity occurs.

Access additional YOLOv5 🚀 resources:

• Wiki – https://github.com/ultralytics/yolov5/wiki
• Tutorials – https://docs.ultralytics.com/yolov5
• Docs – https://docs.ultralytics.com

Access additional Ultralytics ⚡ resources:

• Ultralytics HUB – https://ultralytics.com/hub
• Vision API – https://ultralytics.com/yolov5
• About Us – https://ultralytics.com/about
• Join Our Team – https://ultralytics.com/work
• Contact Us – https://ultralytics.com/contact

Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!

Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐!

[Alberto1404]

@tdhooghe It depends on your application which values work better for you. Default might work for most people but it is not guaranteed that these values are better for all applications. E.g. I use conf_thres=0.6 in my current project because for my problem False Positive is very dangerous. Higher conf_thres will suppress false positive but at the cost of missing some detections and vice versa. It is up to you to decide what better suits to your business problem.

P.S. False positive can also be suppressed by adding more background images but it is just for your understanding. You can always tweak these values to get your desirable outcome.

Hello @creativesalam . Let me know if I am wrong. This is an example training results I obtained for a custom dataset.

Based on these, what conf-thres and iou-thres should I use when running detect.py?