Upload 6DRepNet-HeadPose-OpenVINO: ONNX, OpenVINO IR, Intel NPU blob, card and manifest
ae91bf5 verified | license: mit | |
| base_model: litert-community/6DRepNet-HeadPose-LiteRT | |
| base_model_relation: quantized | |
| library_name: openvino | |
| pipeline_tag: image-classification | |
| tags: | |
| - openvino | |
| - onnx | |
| - intel-npu | |
| # 6DRepNet-HeadPose-OpenVINO | |
| 6DRepNet-HeadPose for Intel NPUs, GPUs and CPUs: 6drepnet from [litert-community/6DRepNet-HeadPose-LiteRT](https://huggingface.co/litert-community/6DRepNet-HeadPose-LiteRT/tree/6a7d8fe0ca042d1524e3912d3371a61d0af45528) as ONNX (tf2onnx 1.17.0, opset 17), OpenVINO IR with fp32 weights, and a precompiled Intel NPU blob. On the NPU of a Core Ultra 9 386H the blob runs in 3.15 ms per run, numpy input to numpy output. | |
| Scripts, raw records, and the same table for all 52 graphs of LiteRT-Models: https://github.com/john-rocky/windows-zoo | |
| ## Files | |
| | file | bytes | sha256 | | |
| |---|---|---| | |
| | `6drepnet.onnx` | 157,356,537 | `7a0ede49b22803e4e6bacbd3549dbde4e4892217022ef63462a974dd452417ea` | | |
| | `openvino/6drepnet.xml` | 295,037 | `2bdc41ebbf65c31529f005aa2503d2ec34155b3dfd31a108a6bc62cbbb082362` | | |
| | `openvino/6drepnet.bin` | 157,293,720 | `d3c69f2b93f72e79044fb61b8f1ebae7bb3d03c3a2c171c3ae4d9c45ea676c50` | | |
| | `npu/6drepnet.blob` | 79,569,025 | `8e31e4cc1fa04d8548894c8e45ba6f00e5c4aa6911f5d8ab37080473c55469ef` | | |
| The NPU blob was compiled with OpenVINO 2026.4.1, Intel NPU driver 32.0.100.4841 (OpenVINO reports NPU_DRIVER_VERSION 1004841, NPU_COMPILER_VERSION 524292). If `import_model` refuses a blob on another driver or OpenVINO version, compile the IR instead (the commented line in the snippet). | |
| ## How it was measured | |
| MSI Stealth 16 AI+ (Core Ultra 9 386H, RTX 5080 Laptop GPU at a 95 W power limit, Windows 11 build 26200.9711, on AC power), 2026-11-07. On the laptop, each row is one process: load, one run, 5 warmup runs, then the median of 30 timed runs (2 warmup + 10 timed runs when the first run takes more than 3 s), numpy input to numpy output, every runtime at its default settings, one process at a time, each started after the CPU, the RTX and the Intel GPU had gone idle (the thresholds are in METHOD.md of the GitHub repo). Output check: each output against the CPU output in fp32 on the same input (ONNX Runtime's CPU EP); PASS = cosine ≥ 0.99, no NaN or Inf, and top-1 agreement ≥ 0.99 for class scores. The Snapdragon rows are Qualcomm AI Hub profile jobs of the same ONNX files (their own timer). | |
| ## Speed and output check | |
| | processor | runtime | file | ms | load s | output check | | |
| |---|---|---|---|---|---| | |
| | Intel AI Boost NPU | OpenVINO 2026.4.1 | `npu/` blob (`import_model`) | 3.15 | 0.05 | PASS (cos ≥ 0.9999) | | |
| | Intel AI Boost NPU | ONNX Runtime 1.31.1 + OpenVINO EP 1.8.1 | `.onnx`, compiled when the session loads | 3.27 | 2.23 | PASS (cos ≥ 0.9999) | | |
| | Intel Graphics (iGPU) | ONNX Runtime 1.31.1 + OpenVINO EP 1.8.1 | `.onnx` | 3.22 | 1.24 | PASS (cos ≥ 0.9999) | | |
| | RTX 5080 Laptop (95 W limit) | ONNX Runtime 1.31.1, CUDA EP | fp16 copy of the `.onnx` (not in this repo) | 2.94 | 1.06 | PASS (cos ≥ 0.9999) | | |
| | RTX 5080 Laptop (95 W limit) | ONNX Runtime 1.31.1, CUDA EP | `.onnx` | 4.30 | 1.26 | PASS (cos ≥ 0.9999) | | |
| | Core Ultra 9 386H CPU | ONNX Runtime 1.31.1, CPU EP | `.onnx` | 16.89 | 0.28 | reference | | |
| | Snapdragon X2 Elite NPU | Qualcomm AI Hub, precompiled QNN ONNX | `.onnx` compiled by AI Hub | 1.27 | 0.37 | not compared (AI Hub profile job, 2026-10-30) | | |
| | Snapdragon X2 Elite NPU | Qualcomm AI Hub, ONNX Runtime QNN EP | `.onnx`, compiled when it loads | 1.26 | 1.57 | not compared (AI Hub profile job, 2026-10-30) | | |
| | Galaxy S26 NPU | Qualcomm AI Hub, precompiled QNN ONNX | `.onnx` compiled by AI Hub | 1.30 | 0.20 | not compared (AI Hub profile job, 2026-10-30) | | |
| The ONNX file against the source `.tflite` (LiteRT interpreter on the CPU, the same input): PASS, cosine ≥ 0.9999 for every output. | |
| ## Run it on the Intel NPU | |
| Download this repo, then run the snippets from its folder: | |
| ```sh | |
| pip install huggingface_hub | |
| hf download mlboydaisuke/6DRepNet-HeadPose-OpenVINO --local-dir 6DRepNet-HeadPose-OpenVINO | |
| cd 6DRepNet-HeadPose-OpenVINO | |
| ``` | |
| OpenVINO (`pip install openvino==2026.4.1`): | |
| ```python | |
| import numpy as np | |
| import openvino as ov | |
| core = ov.Core() | |
| if "NPU" not in core.available_devices: | |
| raise SystemExit(f"no Intel NPU visible to OpenVINO: {core.available_devices}") | |
| with open("npu/6drepnet.blob", "rb") as f: | |
| compiled = core.import_model(f.read(), "NPU") # precompiled: no compile at load | |
| # or compile the IR on this machine: compiled = core.compile_model("openvino/6drepnet.xml", "NPU") | |
| # replace the zeros with your preprocessed input (float32, the shapes printed by compiled.inputs) | |
| feeds = {i.get_any_name(): np.zeros(list(i.shape), np.float32) for i in compiled.inputs} | |
| outputs = compiled(feeds) | |
| print({o.get_any_name(): outputs[o].shape for o in compiled.outputs}) | |
| ``` | |
| ONNX Runtime with the OpenVINO EP (`pip install onnxruntime==1.31.1 onnxruntime-ep-openvino==1.8.1`): | |
| ```python | |
| import numpy as np | |
| import onnxruntime as ort | |
| import onnxruntime_ep_openvino as ov_ep | |
| ort.register_execution_provider_library(ov_ep.get_ep_name(), ov_ep.get_library_path()) | |
| npu = [d for d in ort.get_ep_devices() | |
| if d.ep_name == ov_ep.get_ep_name() and d.ep_metadata.get("ov_device") == "NPU"] | |
| if not npu: | |
| raise SystemExit("no Intel NPU visible to the OpenVINO EP") | |
| so = ort.SessionOptions() | |
| so.add_provider_for_devices(npu, {}) | |
| sess = ort.InferenceSession("6drepnet.onnx", sess_options=so) # compiles for the NPU here | |
| # replace the zeros with your preprocessed input (float32) | |
| feeds = {i.name: np.zeros(i.shape, np.float32) for i in sess.get_inputs()} | |
| outputs = sess.run(None, feeds) | |
| print([o.shape for o in outputs]) | |
| ``` | |
| ## Licence | |
| MIT, the licence of [litert-community/6DRepNet-HeadPose-LiteRT](https://huggingface.co/litert-community/6DRepNet-HeadPose-LiteRT). | |