Skip to content

feat: support weight upload for yolo26-sem semantic segmentation models#492

Merged
leeclemnet merged 3 commits into
mainfrom
feat/yolo26-sem-upload
Jun 9, 2026
Merged

feat: support weight upload for yolo26-sem semantic segmentation models#492
leeclemnet merged 3 commits into
mainfrom
feat/yolo26-sem-upload

Conversation

@leeclemnet

@leeclemnet leeclemnet commented Jun 9, 2026

Copy link
Copy Markdown
Contributor

Description

Prior behavior: The SDK had no awareness of semantic segmentation as a
distinct task. TYPE_SEMANTIC_SEGMENTATION projects were explicitly skipped
in upload validation, and SemanticSegmentationModel checkpoints (new in
ultralytics 8.4.55) were unrecognised.

New behavior: A new TASK_SEM = "sem" constant is wired through the
weight upload pipeline:

  • task_of_model_type("yolo26-sem")TASK_SEM
  • _detect_yolo_task() recognises SemanticSegmentationModel checkpoints
  • validate_model_type_for_project() enforces that -sem models match
    semantic-segmentation projects (and rejects mismatches in both directions)
  • Auto-suffix: declaring plain "yolo26" with a semantic-seg checkpoint
    auto-corrects to "yolo26-sem"

No changes needed to supported_models, dependency checks, or artifact
extraction — "yolo26" in "yolo26-sem" passes all existing gates.

Motivation: roboflow-train and roboflow-model-conversion already support
yolo26-sem; the SDK was the missing link for user-side weight upload.

Type of change

  • New feature (non-breaking change which adds functionality)

How has this change been tested, please provide a testcase or example of how you tested the change?

  • unit tests
  • versionless weight upload (pre-trained cityscales yolo26n-sem) succeeds on prod
image - [x] versioned weight upload (same model) succeeds on staging with https://github.com/roboflow/roboflow/pull/12297 image

Will the change affect Universe? If so was this change tested in universe?

n/a

Github Actions Required to Deploy

n/a — SDK package, no deployment needed.

Infrastructure impact

  • This change affects infrastructure needs (GPUs, Cloud Function sizing, storage etc.)
  • This change requires updates to secrets
  • This change materially impacts cloud costs (explain briefly)

Docs

  • This change requires a documentation update
  • Docs updated? What were the changes:

Add TASK_SEM constant and wire it through the weight upload pipeline so
yolo26-sem checkpoints (ultralytics SemanticSegmentationModel) are
recognised, auto-suffixed, and validated against semantic-segmentation
projects.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@leeclemnet leeclemnet marked this pull request as ready for review June 9, 2026 14:52
@leeclemnet leeclemnet requested a review from probicheaux June 9, 2026 15:18
leeclemnet and others added 2 commits June 9, 2026 13:24
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@leeclemnet leeclemnet merged commit 107fb34 into main Jun 9, 2026
15 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants