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

Fix Keypoint & Instance Segmentation Training Hyperparameters #163

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

Fix Keypoint & Instance Segmentation Training Hyperparameters #163

wants to merge 7 commits into from

Conversation

JSabadin
Copy link
Contributor

@JSabadin JSabadin commented Jan 29, 2025 •
edited
Loading

🔥 Updated Hyperparameters

What's New?

  • Added optimized hyperparameters for keypoint detection and instance segmentation
  • Experimentally determined for best performance .
  • Tested on Tiger and Crack datasets, achieving SOTA performance 🚀

Results

image
image

@JSabadin JSabadin requested a review from a team as a code owner January 29, 2025 07:08
@JSabadin JSabadin requested review from kozlov721, klemen1999, tersekmatija and conorsim and removed request for a team January 29, 2025 07:08
@github-actions github-actions bot added documentation Improvements or additions to documentation fix Fixing a bug labels Jan 29, 2025
klemen1999
klemen1999 reviewed Jan 29, 2025
View reviewed changes
Copy link
Collaborator

@klemen1999 klemen1999 left a comment

Choose a reason for hiding this comment

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

Generally LGTM, just left some comments

configs/instance_segmentation_light_model.yaml Outdated Show resolved Hide resolved
luxonis_train/core/utils/train_utils.py Show resolved Hide resolved
kozlov721
kozlov721 approved these changes Jan 29, 2025
View reviewed changes
Copy link
Collaborator

@kozlov721 kozlov721 left a comment

Choose a reason for hiding this comment

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

LGTM, just left a small comment

luxonis_train/config/config.py Show resolved Hide resolved
luxonis_train/config/config.py Outdated Show resolved Hide resolved
@JSabadin JSabadin requested a review from klemen1999 January 30, 2025 12:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@kozlov721 kozlov721 kozlov721 approved these changes

@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

@klemen1999 klemen1999 Awaiting requested review from klemen1999

Assignees

@JSabadin JSabadin

Labels
documentation Improvements or additions to documentation fix Fixing a bug
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@JSabadin @kozlov721 @klemen1999