getitune-running-inference
Compare original and translation side by side
🇺🇸
Original
English🇨🇳
Translation
ChineseRunning inference with getitune
使用getitune运行推理
getituneengine.predict()engine.test()model=Run everything from .
library/getituneengine.predict()engine.test()model=请在目录下运行所有命令。
library/PyTorch inference (trained model)
PyTorch推理(训练后模型)
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 subsetpython
from getitune.engine import create_engine
engine = create_engine(
model="efficientnet_b0",
data="/path/to/dataset",
)
test_metrics = engine.test() # 测试子集的指标计算
predictions = engine.predict() # 测试子集的预测结果OpenVINO / ONNX inference (exported model)
OpenVINO / ONNX推理(导出后模型)
python
from getitune.engine import create_enginepython
from getitune.engine import create_engineOpenVINO IR — pass the .xml
OpenVINO IR — 传入.xml文件路径
ov_engine = create_engine(model="/path/to/exported_model.xml", data="/path/to/dataset")
ov_engine.test()
ov_engine.predict()
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 — 传入.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).onnx_engine = create_engine(model="/path/to/exported_model.onnx", data="/path/to/dataset")
onnx_engine.test()
onnx_engine.predict()
传入`.xml`或`.onnx`路径会构建一个`OVEngine`,它将通过[ModelAPI](https://github.com/open-edge-platform/model_api)加载模型。Workflow
工作流程
- Pick the model surface. Use a model name/checkpoint for PyTorch inference,
or an exported /
.xmlfor deployed inference..onnx- Done when: returns the expected engine type.
create_engine(...)
- Done when:
- Point at a dataset with a test subset (see
data=).getitune-preparing-datasets- Done when: runs without a data/format error.
engine.test()
- Done when:
- Run for metrics or
test()for per-item outputs.predict()- Done when: metrics are produced, or predictions are returned for each item.
- 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.
- 选择模型形式:使用模型名称/checkpoint进行PyTorch推理,或使用导出的/
.xml文件进行部署式推理。.onnx- 完成标志:返回预期的引擎类型。
create_engine(...)
- 完成标志:
- 将指向包含测试子集的数据集(参考
data=)。getitune-preparing-datasets- 完成标志:运行时无数据/格式错误。
engine.test()
- 完成标志:
- 运行获取指标或
test()获取逐样本输出。predict()- 完成标志:生成指标,或返回每个样本的预测结果。
- 验证导出模型时对比不同后端:PyTorch与OpenVINO/ONNX的指标应高度匹配(允许微小数值偏差)。
- 完成标志:导出模型的指标与PyTorch模型的指标在误差容忍范围内。
CLI
命令行界面(CLI)
bash
undefinedbash
undefinedfrom library/
在library/目录下执行
getitune predict --data_root /path/to/dataset --model efficientnet_b0
getitune test --data_root /path/to/dataset --model /path/to/exported_model.xml
undefinedgetitune predict --data_root /path/to/dataset --model efficientnet_b0
getitune test --data_root /path/to/dataset --model /path/to/exported_model.xml
undefinedRelated skills
相关技能
- — produce the
getitune-exporting-a-model/.xmlused here..onnx - — run inference with an INT8 quantized model.
getitune-optimizing-a-model - — the
getitune-preparing-datasetshalf of inference.data=
- — 生成本文中使用的
getitune-exporting-a-model/.xml文件。.onnx - — 使用INT8量化模型运行推理。
getitune-optimizing-a-model - — 推理流程中
getitune-preparing-datasets相关的部分。data=