Skip to content

Commit 76ede38

Browse files
committed
feat: ONNX Runtime + CoreML EP for YOLO detection on Apple Silicon
- Ship yolo26n.onnx (9.4MB) from onnx-community for zero-config default - On-demand download for s/m/l from onnx-community HuggingFace repos - _OnnxCoreMLModel wrapper: logits+pred_boxes → sigmoid+cxcywh→xyxy - Bypasses macOS 26.x MPSGraph MLIR crash (SIGABRT) - 15-25ms/frame (~67 FPS) on Apple M5 Pro with CoreML EP - Strip requirements_mps.txt: no torch/ultralytics (~120MB → ~17MB) - Class names from yolo26n_names.json (no .pt dependency)
1 parent a68cd16 commit 76ede38

File tree

5 files changed

+663
-53
lines changed

5 files changed

+663
-53
lines changed
Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,8 @@
11
# YOLO 2026 — MPS (Apple Silicon) requirements
2-
# Standard PyTorch — MPS backend is included by default on macOS
3-
torch>=2.4.0
4-
torchvision>=0.19.0
5-
ultralytics>=8.3.0
6-
coremltools>=8.0
2+
# Uses ONNX Runtime + CoreML EP for GPU/ANE acceleration.
3+
# Pre-built yolo26n.onnx is shipped in the repo, so torch/ultralytics
4+
# are NOT needed at runtime — only onnxruntime for inference.
5+
onnxruntime>=1.19.0
76
numpy>=1.24.0,<2.0.0
87
opencv-python-headless>=4.8.0
98
Pillow>=10.0.0
10-

0 commit comments

Comments
 (0)