Skip to main content

getitune-running-inference

Run inference and evaluation with a getitune model (the Geti training library). Use when a user wants to call `engine.predict()` / `engine.test()` or `getitune predict` / `getitune test`, run inference with a PyTorch checkpoint versus an exported OpenVINO IR (`.xml`) or ONNX (`.onnx`) model, or understand how `OVEngine` loads deployed models via ModelAPI. Covers PyTorch, OpenVINO, and ONNX inference backends.

معلومات المصدر

المستودع
open-edge-platform/geti
آخر نشاط في المصدر
٢٥ أغسطس ٢٠٢٦ في ٠٩:٢٩
لغة SKILL.md المكتشفة
الإنجليزية
النجوم
١٬٣٥٢
التفرعات
٤٨٠

خيارات التثبيت

يُحدَّد Prompt الذي يراجع المصدر أولًا بشكل افتراضي. يمكنك التبديل إلى أمر مباشر أو تنزيل نسخة محلية.

مراجعة ملفات المصدر

اقرأ SKILL.md وأي ملفات مرافقة يعرضها SkillsMP قبل أن تقرر التثبيت.

عرض SKILL.md

SKILL.md
تعليمات المصدر · معاينة للقراءة فقط
name
getitune-running-inference
description
Run inference and evaluation with a getitune model (the Geti training library). Use when a user wants to call `engine.predict()` / `engine.test()` or `getitune predict` / `getitune test`, run inference with a PyTorch checkpoint versus an exported OpenVINO IR (`.xml`) or ONNX (`.onnx`) model, or understand how `OVEngine` loads deployed models via ModelAPI. Covers PyTorch, OpenVINO, and ONNX inference backends.
# Running inference with getitune `getitune` runs inference through `engine.predict()` (per-item predictions) and `engine.test()` (metrics on the test subset). The same calls work whether the engine holds a **PyTorch** model or an **exported** OpenVINO/ONNX model — the backend is selected from what you pass to `model=`. Run everything from `library/`. ## PyTorch inference (trained model) ```python from getitune.engine import create_engine engine = create_engine( model="efficientnet_b0", data="/path/to/dataset", ) test_metrics = engine.test() # metrics on the test subset predictions = engine.predict() # predictions on the test subset ``` ## OpenVINO / ONNX inference (exported model) ```python from getitune.engine import create_engine # OpenVINO IR — pass the .xml ov_engine = create_engine(model="/path/to/exported_model.xml", data="/path/to/dataset") ov_engine.test() ov_engine.predict() # ONNX — pass the .onnx onnx_engine = create_engine(model="/path/to/exported_model.onnx", data="/path/to/dataset") onnx_engine.test() onnx_engine.predict() ``` Passing an `.xml` or `.onnx` path builds an `OVEngine`, which loads the model via [ModelAPI](https://github.com/open-edge-platform/model_api). ## Workflow 1. **Pick the model surface.** Use a model name/checkpoint for PyTorch inference, or an exported `.xml`/`.onnx` for deployed inference. - Done when: `create_engine(...)` returns the expected engine type. 2. **Point `data=` at a dataset with a test subset** (see `getitune-preparing-datasets`). - Done when: `engine.test()` runs without a data/format error. 3. **Run `test()` for metrics or `predict()` for per-item outputs.** - Done when: metrics are produced, or predictions are returned for each item. 4. **Compare backends when validating an export.** PyTorch vs OpenVINO/ONNX metrics should closely match (small numeric drift is expected). - Done when: exported-model metrics are within tolerance of the PyTorch model. ## CLI ```bash # from library/ getitune predict --data_root /path/to/dataset --model efficientnet_b0 getitune test --data_root /path/to/dataset --model /path/to/exported_model.xml ``` ## Related skills - `getitune-exporting-a-model` — produce the `.xml`/`.onnx` used here. - `getitune-optimizing-a-model` — run inference with an INT8 quantized model. - `getitune-preparing-datasets` — the `data=` half of inference.
عرض على GitHub