-
Notifications
You must be signed in to change notification settings - Fork 17
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
weight design #36
Comments
The paper mentions that the calculation of t is composed of s and IoU, but in the code implementation, the calculation of t is obtained by multiplying s with the predicted location (loc). Specifically, the code implementation multiplies s with the predicted location (loc) to obtain a new vector, which is then passed through the sigmoid function to obtain t. |
In the code, t = p_cls * p_loc. p_loc = torch.exp(-5reg_loss) = torch.exp(-5 (-torch.log(iou))) = iou5 if we use IoU loss. So t = p_cls * iou5 which is the same to the paper. t is not then passed through sigmoid function. |
Did I not understand the paper? Why is the Loss function in the code very different from that in the paper?
cls_loss in the code is only obtained by p_pos_weight weight, but this is not the case in the paper.
The text was updated successfully, but these errors were encountered: