-
Notifications
You must be signed in to change notification settings - Fork 1.8k
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
feat: 🚀 initial keypoint support for transformers added #1553
base: develop
Are you sure you want to change the base?
Conversation
Signed-off-by: Onuralp SEZER <thunderbirdtr@gmail.com>
@merveenoyan 👋 hello I notice current latest and dev doesn't have "post_process_keypoint_detection" function and I was able to run via only this pull request branch "huggingface/transformers#33200" Today I read about "https://huggingface.co/docs/transformers/tasks/keypoint_detection" you posted in social and I was unable run keypoint but I was able to do via using install this repo/branch, I think it is forgot to merge ?
Error output when I ran from "main"
Here is my initial collab link for working version https://colab.research.google.com/drive/1gxpE9iDR7gXl2VwwScB5W8Yw-e_xUo3i?usp=sharing |
@LinasKo missing function basically converting normalize values to w,h values to position correctly should I finish based on that in current release and finish the PR ? |
@onuralpszr, I need more details.
|
Sorry, I was little bit hasty I guess and for answering your questions 1 - Missing function is theirs and it is post process function for converting normalize xy values to pixel based on xy values (post_process_keypoint_detection is missing) |
@LinasKo I see approve so I am going to move forward based on this PR ("huggingface/transformers#33200" ) I am going to finish this PR |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I added a few small comments. Let's wait until they merge the change, test it in a Colab, make sure docs are available via mkdocs serve
and then merge it!
supervision/keypoint/core.py
Outdated
return cls( | ||
xy=np.array(keypoints_list), | ||
confidence=np.array(scores_list), | ||
class_id=None, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Do we need to explicitly set it to None
?
…and class_id removed Signed-off-by: Onuralp SEZER <thunderbirdtr@gmail.com>
dfa7ff3
to
540516d
Compare
Signed-off-by: Onuralp SEZER <thunderbirdtr@gmail.com>
540516d
to
ed928d2
Compare
Hey @onuralpszr , How's this one going? Is it ready for review, or is there some work left to do? |
I have small problem when I detect multiple images at once keypoint list doesn't come in same sizes. Point lists are different |
Description
Recent release of transformers introduce keypoint support and initial support for keypoint added.
Type of change
Docs