Skip to content

YOLO 模块(yolo)

yolo 模块约定

对当前屏截图检测;区域四元组在前。无默认值,参数均必填。

最短示例:

javascript
yolo.loadModel('model.mlmodel');
const hits = yolo.findClass('button', 0.5, 0.45, 10);
if (hits[0]) auto.click(hits[0].centerX, hits[0].centerY);

yolo.loadModel / getModelInfo

typescript
function yolo.loadModel(modelPath: string): Promise<YoloModelInfo>;
function yolo.getModelInfo(): Promise<YoloModelInfo>;

返回字段:

字段类型描述
loadedboolean是否已加载
model / modelPathstring模型名 / 路径
classCountnumber类别数
labelsstring[]类别名

yolo.detect - 目标检测

typescript
function yolo.detect(confidence, iou, limit): Promise<YoloDetection[]>;
function yolo.detect(x1,y1,x2,y2, confidence, iou, limit): Promise<YoloDetection[]>;

参数说明:

参数名类型是否必填默认值描述
confidencenumber-置信度阈值
iounumber-IoU 阈值
limitnumber-最多检测数

返回字段:

字段类型描述
label / classIndex / confidence-类别与置信度
x,y,ex,ey,width,height,centerX,centerYnumber

示例:

javascript
let dets = yolo.detect(0.5, 0.45, 100);
const region = yolo.detect(0, 0, 400, 600, 0.5, 0.45, 100);

yolo.findClass - 按类名查找

typescript
function yolo.findClass(className, confidence, iou, limit): Promise<YoloDetection[]>;
function yolo.findClass(x1,y1,x2,y2, className, confidence, iou, limit): Promise<YoloDetection[]>;

参数说明:

参数名类型是否必填默认值描述
classNamestring-类别名
confidencenumber-置信度阈值
iounumber-IoU 阈值
limitnumber-最多命中数

示例:

javascript
let hits = yolo.findClass('person', 0.5, 0.45, 10);