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

Add Keypoint and Instance Segmentation support in visualizer #261

Merged

Conversation

ashwinvaidya17
Copy link
Collaborator

@ashwinvaidya17 ashwinvaidya17 commented Jan 24, 2025

  • Introduced Keypoint primitive for visualizing keypoints on images.
  • Updated visualizer to handle Instance Segmentation results and added corresponding scene.
  • Refactored existing scenes to integrate new functionalities, including KeypointScene and InstanceSegmentationScene.
  • Enhanced Polygon primitive with opacity and outline width options.
  • Updated tests to cover new features and ensure functionality.

This commit enhances the visualizer's capabilities for handling keypoints and instance segmentation, improving the overall visualization experience.

What does this PR do?

MaskRCNN with XAI Head

image

RTM Pose

image

Fixes # (issue)

Before submitting

  • This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
  • Did you make sure to update the documentation with your changes?
  • Did you write any new necessary tests?

Sorry, something went wrong.

Verified

This commit was signed with the committer’s verified signature. The key has expired.
dtzWill Will Dietz
- Introduced Keypoint primitive for visualizing keypoints on images.
- Updated visualizer to handle Instance Segmentation results and added corresponding scene.
- Refactored existing scenes to integrate new functionalities, including KeypointScene and InstanceSegmentationScene.
- Enhanced Polygon primitive with opacity and outline width options.
- Updated tests to cover new features and ensure functionality.

This commit enhances the visualizer's capabilities for handling keypoints and instance segmentation, improving the overall visualization experience.
@github-actions github-actions bot added tests Related to tests python python related changes labels Jan 24, 2025
@ashwinvaidya17 ashwinvaidya17 marked this pull request as ready for review January 24, 2025 12:40
Copy link
Collaborator

@sovrasov sovrasov left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the PR! Could you add visualization to examples/python?

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
Signed-off-by: Ashwin Vaidya <ashwinnitinvaidya@gmail.com>
Signed-off-by: Ashwin Vaidya <ashwinnitinvaidya@gmail.com>
@github-actions github-actions bot added the docs Related to documentation label Jan 27, 2025
@sovrasov sovrasov added this pull request to the merge queue Jan 27, 2025
Merged via the queue into openvinotoolkit:master with commit db0bf5a Jan 27, 2025
16 checks passed
@sovrasov sovrasov added this to the 0.2.6 milestone Jan 28, 2025
layout=layout,
)

def _get_keypoints(self, result: DetectedKeypoints) -> list[Keypoint]:
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@ashwinvaidya17, would it be an idea to use predictions instead of results? What is results object? A prediction? Or something else?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
docs Related to documentation python python related changes tests Related to tests
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants