Views
No views yet
icon)1import mlx.core as mx
2from mlx_vlm.models.yolo11 import YOLO11, load_weights, non_max_suppression
3
4model = YOLO11(nc=1)
5load_weights(model, mx.load("model.safetensors"))
6
7# Run detection
8pred = model(mx.array(image)[None]) # (1, 5, N) — 4 box + 1 score
9detections = non_max_suppression(pred, conf_thresh=0.05, iou_thresh=0.1)microsoft/OmniParser-v2.0/icon_detect/model.pt (Ultralytics YOLO11, nc=1).
Parity verified: identical detections vs PyTorch reference (7/7 boxes, confidences to 4 decimals).