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

Why is precision so low for yolo-nas model ? #1191

Closed
HissaanAli opened this issue Jun 19, 2023 · 1 comment
Closed

Why is precision so low for yolo-nas model ? #1191

HissaanAli opened this issue Jun 19, 2023 · 1 comment

Comments

@HissaanAli
Copy link

💡 Your Question

I trained a yolo-nas model for an object detection dataset and I noticed something weird. The recall and mAP value of the model were at 98% and 95% respectively however the precision was barely 20%.

This comes off as a surprise to me that how is the mAP and recall high, but the precision so low. Why is that ? Can somebody elaborate ?

The same model when trained with yolov5 gave precision, recall and mAP value of 94%.

Versions

No response

@ofrimasad
Copy link
Collaborator

Hi @HissaanAli .
It sounds like you need to tune the confidence level of the NMS.
In the post prediction callback, pass conf=0.9, for example. And see if the precision-recall tradeoff looks better

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

No branches or pull requests

3 participants