Skip to content

模型推理

使用已加载的 ONNX 模型对指定设备的进行 AI 推理预测,返回模型的推理结果。

接口说明

接口类型

predictOnnxModel

参数

参数名类型必填说明
onnxidstring通过 loadOnnxModel 返回的模型 ID
deviceIdstring设备 ID,用于获取屏幕截图
rectarray预测区域 [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);

商务合作:try.catch@foxmail.com