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

作者您好,请问yolo_loss中,怎么计算的loss confidence的呀? #38

Open
ZhongyuanW opened this issue Dec 4, 2020 · 1 comment

Comments

@ZhongyuanW
Copy link

作者您好,您的代码写得太好了!我在阅读yolo_loss的时候有个疑问,您loss confidence 是怎么计算的呢?特别是iou_max是代表的什么含义呀?
image

@wangren97
Copy link

hi, first line means check the IOU between ground truth bounding boxes and predicted ones. second line means find the maximum IOU. third line means if the max IOU is smaller than the predefined threshold, then consider them as no object ones. so finally calculate the loss as object + no object. object: the objectness score should be 1. no object: the objectness score should be 0. The loss follows focal loss type instead of BCE loss.

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