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

About the LengthIoU loss #1

Closed
gintamatyy opened this issue Feb 28, 2023 · 4 comments
Closed

About the LengthIoU loss #1

gintamatyy opened this issue Feb 28, 2023 · 4 comments

Comments

@gintamatyy
Copy link

Congratulations on the TIP publication, and thanks for sharing your code. I am interested in this method, especially the LengthIoU loss in your paper. But I did not find the code for it. Could you show me the implementation of this loss or tell me the path of the code if I missed it?

@ljn114514
Copy link
Owner

You can find it at line 170-175 in models/kpt_head.py

@gintamatyy
Copy link
Author

Thank you for your reply! While I found PolarIoU loss rather than LengthIoU loss here. Did the gt_lens* cos(δθ) finish before kpt_head?

@ljn114514
Copy link
Owner

The line 170-175 in models/kpt_head.py corresponding to equal 10 in the paper. If you mean equal 11, you can find it at line 367-374.(replace line 361-365)

@gintamatyy
Copy link
Author

Got it! Thanks for your help!

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