Skip to content

Feat/yolo 26 integration #200

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

Merged
dtronmans merged 18 commits into from
Feb 6, 2026
Merged

Feat/yolo 26 integration #200

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
ba1d9fd
Working object detection changes
dtronmans Jan 29, 2026
21af145
Check end2end mode detection
dtronmans Jan 29, 2026
94b9e7d
Yolo26 heads with one output (minus topK and conf thresholding) for d…
dtronmans Feb 2, 2026
a127b67
Update comments, only support Proto26
dtronmans Feb 3, 2026
8edc34b
cleanup, seg pose tests
dtronmans Feb 3, 2026
4a8ca16
Revert hard-coding parser to YOLOExtendedParser
dtronmans Feb 3, 2026
922b832
pre-commit
dtronmans Feb 3, 2026
d3e7f6e
Complete reversal OBB
dtronmans Feb 3, 2026
eab01cc
[Automated] Updated coverage badge
actions-user Feb 3, 2026
396bd96
precommit again
dtronmans Feb 3, 2026
251381b
Merge branch 'feat/yolo-26-integration' of https://github.com/luxonis…
dtronmans Feb 3, 2026
356a2aa
Correct download links for weights
dtronmans Feb 3, 2026
0c92996
precommit
dtronmans Feb 4, 2026
b38fd5b
working yolo26 separate exporter head
dtronmans Feb 5, 2026
897eebb
revert yolo8 (try2) to main and fix yolo26
dtronmans Feb 5, 2026
6533246
More correct usage of fuse, revert yolov8
dtronmans Feb 5, 2026
a11dc17
Temporary pass YOLOExtendedParser to detect yolo26
dtronmans Feb 5, 2026
c16246c
Comments slightly more accurate about output mask
dtronmans Feb 5, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions media/coverage_badge.svg
View file
Open in desktop
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
14 changes: 13 additions & 1 deletion tests/conftest.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,19 @@ def pytest_addoption(parser):
)
parser.addoption(
"--yolo-version",
choices=["v5", "v6", "v6r2", "v6r4", "v7", "v8", "v9", "v10", "v11", "v12"],
choices=[
"v5",
"v6",
"v6r2",
"v6r4",
"v7",
"v8",
"v9",
"v10",
"v11",
"v12",
"v26",
],
default=None,
help="If set then test only that specific yolo version",
)
Expand Down
14 changes: 14 additions & 0 deletions tests/constants.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -69,6 +69,13 @@
{"name": "yolov12m", "version": "v12"},
{"name": "yolov12l", "version": "v12"},
{"name": "yolov12x", "version": "v12"},
{"name": "yolo26n", "version": "v26"},
{"name": "yolo26s", "version": "v26"},
{"name": "yolo26m", "version": "v26"},
{"name": "yolo26l", "version": "v26"},
{"name": "yolo26x", "version": "v26"},
{"name": "yolo26n-seg", "version": "v26"},
{"name": "yolo26n-pose", "version": "v26"},
{"name": "yolov8n-cls", "version": "v8"},
{"name": "yolov8n-seg", "version": "v8"},
{"name": "yolov8n-pose", "version": "v8"},
Expand Down Expand Up @@ -157,6 +164,13 @@
"yolov12m": "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo12m.pt",
"yolov12l": "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo12l.pt",
"yolov12x": "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo12x.pt",
"yolo26n": "https://github.com/ultralytics/assets/releases/download/v8.4.0/yolo26n.pt",
"yolo26s": "https://github.com/ultralytics/assets/releases/download/v8.4.0/yolo26s.pt",
"yolo26m": "https://github.com/ultralytics/assets/releases/download/v8.4.0/yolo26m.pt",
"yolo26l": "https://github.com/ultralytics/assets/releases/download/v8.4.0/yolo26l.pt",
"yolo26x": "https://github.com/ultralytics/assets/releases/download/v8.4.0/yolo26x.pt",
"yolo26n-seg": "https://github.com/ultralytics/assets/releases/download/v8.4.0/yolo26n-seg.pt",
"yolo26n-pose": "https://github.com/ultralytics/assets/releases/download/v8.4.0/yolo26n-pose.pt",
"yolov11n-cls": "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-cls.pt",
"yolov11n-seg": "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-seg.pt",
"yolov11n-pose": "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-pose.pt",
Expand Down
6 changes: 6 additions & 0 deletions tools/main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@
YOLOV10_CONVERSION,
YOLOV11_CONVERSION,
YOLOV12_CONVERSION,
YOLOV26_CONVERSION,
detect_version,
)

Expand All @@ -48,6 +49,7 @@
YOLOV10_CONVERSION,
YOLOV11_CONVERSION,
YOLOV12_CONVERSION,
YOLOV26_CONVERSION,
]


Expand Down Expand Up @@ -178,6 +180,10 @@ def convert(
from tools.yolo.yolov8_exporter import YoloV8Exporter

exporter = YoloV8Exporter(str(model_path), config.imgsz, config.use_rvc2)
elif version == YOLOV26_CONVERSION:
from tools.yolo.yolo26_exporter import Yolo26Exporter

exporter = Yolo26Exporter(str(model_path), config.imgsz, config.use_rvc2)
elif version == YOLOV10_CONVERSION:
from tools.yolo.yolov10_exporter import YoloV10Exporter

Expand Down
6 changes: 6 additions & 0 deletions tools/modules/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,11 @@
DetectV7,
DetectV8,
DetectV10,
DetectV26,
PoseV8,
PoseV26,
SegmentV8,
SegmentV26,
)
from .stage2 import Multiplier

Expand All @@ -27,11 +30,14 @@
"DetectV8",
"Exporter",
"PoseV8",
"PoseV26",
"OBBV8",
"SegmentV8",
"SegmentV26",
"ClassifyV8",
"Multiplier",
"DetectV5",
"DetectV7",
"DetectV10",
"DetectV26",
]
Loading
Loading