Skip to content

Conversation

@kozlov721
Copy link
Collaborator

@kozlov721 kozlov721 commented Jan 14, 2025

Integration of changes in luxonis-ml introduced in #211.

Config Changes

  • Field task in node configs renamed to task_name for naming consistency with luxonis-ml

Example
Let's have a dataset with 3 tasks; motorbike and car for detection, and vehicles for semantic segmentation. The labels would have the following keys:

{
    "motorbike/boundingbox",
    "motorbike/keypoints",
    "car/boundingbox",
    "car/keypoints",
    "vehicles/segmentation",
}

The following configuration for luxonis-train would make use of the above tasks:

model:
  name: parking_lot_model
  nodes:

    - name: EfficientRep

    - name: RepPANNeck

    - name: EfficientKeypointBBoxHead
      task_name: motorbike

    - name: EfficientKeypointBBoxHead
      task_name: car

    - name: SegmentationHead
      task_name: vehicles

@kozlov721 kozlov721 requested a review from a team as a code owner January 14, 2025 15:11
@kozlov721 kozlov721 requested review from conorsim, klemen1999 and tersekmatija and removed request for a team January 14, 2025 15:11
@github-actions github-actions bot added documentation Improvements or additions to documentation enhancement New feature or request CLI Changes affecting the CLI labels Jan 14, 2025
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, left some comments

Copy link
Contributor

@sokovninn sokovninn left a comment

Choose a reason for hiding this comment

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

LGTM

Copy link
Contributor

@JSabadin JSabadin left a comment

Choose a reason for hiding this comment

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

LGTM!

@kozlov721 kozlov721 merged commit f5d9a03 into main Jan 22, 2025
5 of 6 checks passed
@kozlov721 kozlov721 deleted the feature/nested-labels branch January 22, 2025 04:47
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLI Changes affecting the CLI documentation Improvements or additions to documentation enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants