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

Threshold option during detection #49

Closed
SlavaKeshkov opened this issue Aug 11, 2019 · 1 comment
Closed

Threshold option during detection #49

SlavaKeshkov opened this issue Aug 11, 2019 · 1 comment

Comments

@SlavaKeshkov
Copy link

Hey @zzh8829, thanks for your code, works great. I was thinking it would be a good idea to have a detection threshold flag --thresh for the detection.py or conversion.py. Do I get it right that iou_threshold and score_threshold in lines 190-192 of models.py is the only way to change threshold for prediction? Let me know, this flag along with num_classes flag would be a great boost to the repo.

@zzh8829
Copy link
Owner

zzh8829 commented Aug 14, 2019

done

@zzh8829 zzh8829 closed this as completed Aug 14, 2019
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

2 participants