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.

Sign up for GitHub

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

Jump to bottom

Instance Segmentation Head #144

Open
wants to merge 9 commits into
base: main
Choose a base branch
from
Open

Instance Segmentation Head #144

wants to merge 9 commits into from

Conversation

JSabadin
Copy link
Contributor

New Instance Segmentation Loss and Related Enhancements

Description

In this PR, we propose:

  • New Detection Head: Enhanced architecture for improved performance.
  • New Segmentation Head: Built on top of the detection head to support instance segmentation.
  • New Loss Functions:
    • Detection loss: Optimized for precision.
    • Segmentation loss: Designed for instance segmentation.
  • New Visualizer: Provides visualization of instance segmentation training results.
  • Extended mAP Metric: Now supports instance segmentation evaluation.
  • Luxonis-ML Loader Support: Added compatibility with the new loader.

Overfitting Examples

example2 example1 example3

@JSabadin JSabadin requested a review from a team as a code owner December 10, 2024 13:35
@JSabadin JSabadin requested review from kozlov721, klemen1999, tersekmatija and conorsim and removed request for a team December 10, 2024 13:35
@github-actions github-actions bot added the enhancement New feature or request label Dec 10, 2024
@kozlov721 kozlov721 changed the title Feat/instance seg head Instance Segmentation Head Dec 10, 2024
@github-actions github-actions bot added the documentation Improvements or additions to documentation label Dec 11, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@kozlov721 kozlov721 Awaiting requested review from kozlov721 kozlov721 is a code owner automatically assigned from luxonis/ml-reviewers

@klemen1999 klemen1999 Awaiting requested review from klemen1999 klemen1999 is a code owner automatically assigned from luxonis/ml-reviewers

@tersekmatija tersekmatija Awaiting requested review from tersekmatija tersekmatija is a code owner automatically assigned from luxonis/ml-reviewers

@conorsim conorsim Awaiting requested review from conorsim conorsim is a code owner automatically assigned from luxonis/ml-reviewers

At least 1 approving review is required to merge this pull request.

Assignees

@JSabadin JSabadin

Labels
documentation Improvements or additions to documentation enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@JSabadin