Skip to content

YOLO26 ReduceMax in graph + YOLO26 with NMS in graph option#206

Closed
dtronmans wants to merge 4 commits intomainfrom
feat/yolo26-reducemax
Closed

YOLO26 ReduceMax in graph + YOLO26 with NMS in graph option#206
dtronmans wants to merge 4 commits intomainfrom
feat/yolo26-reducemax

Conversation

@dtronmans
Copy link
Contributor

Purpose

Was just a bit worried about clashes with the PR that adds Yolo26 NMS so I merged the two in this PR.

Yolo26 conversion with one2one head:

  • tools yolo26n.pt --imgsz "640" --version yolov26_nms
  • Since this is the first time the same .pt file can have two different export branches, I had to add cli_version flag to the tests

ReduceMax in graph:

  • Not an option/flag it's just always there
  • Need a corresponding PR in depthai-nodes that does not perform cls_scores = scores.max(axis=-1)

Specification

Dependencies & Potential Impact

Deployment Plan

Testing & Validation

Detect:
detect26

Pose:
pose26

Segment:
seg26

@github-actions
Copy link

☂️ Python Coverage

current status: ✅

Overall Coverage

Lines Covered Coverage Threshold Status
30374 6498 21% 0% 🟢

New Files

No new covered files...

Modified Files

File Coverage Status
tools/main.py 86% 🟢
tools/modules/heads.py 51% 🟢
tools/version_detection/version_detection.py 47% 🟢
tools/yolo/yolov8_exporter.py 64% 🟢
TOTAL 62% 🟢

updated for commit: a854110 by action🐍

@aljazkonec1
Copy link

Thanks for the quick implementation! Before merge I want to test out parser implementation on depthai to make sure no no other changes to the model are needed, if thats OK @klemen1999

@github-actions
Copy link

Test Results

  6 files    6 suites   23m 16s ⏱️
 26 tests  26 ✅ 0 💤 0 ❌
156 runs  156 ✅ 0 💤 0 ❌

Results for commit a854110.

@dtronmans dtronmans closed this Feb 25, 2026
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