模型推理
使用已加载的 ONNX 模型对指定设备的进行 AI 推理预测,返回模型的推理结果。
接口说明
接口类型
predictOnnxModel参数
| 参数名 | 类型 | 必填 | 说明 |
|---|---|---|---|
| onnxid | string | 是 | 通过 loadOnnxModel 返回的模型 ID |
| deviceId | string | 是 | 设备 ID,用于获取屏幕截图 |
| rect | array | 否 | 预测区域 [x1, y1, x2, y2],默认全屏 |
返回值
返回模型的推理结果。
javascript
// YOLO 目标检测模型的典型返回格式
[
{
"classId": 0,
"x": 100,
"y": 200,
"confidence": 0.95,
"boundingBox": { ... } // [x, y, width, height]
},
{
"classId": 1,
"x": 100,
"y": 200,
"confidence": 0.87,
"boundingBox": { ... }
}
]基本使用
javascript
// 1. 加载模型
const onnxid = await apiInvoke('loadOnnxModel', {
file: 'C:\\models\\yolov5s.onnx'
});
// 2. 进行推理
const result = await apiInvoke('predictOnnxModel', {
onnxid: onnxid,
deviceId: 'P72578581E07'
});
console.log('检测结果:', result);